首页 > 代码库 > bzoj 2038 莫队入门
bzoj 2038 莫队入门
http://www.lydsy.com/JudgeOnline/problem.php?id=2038
题意:多次询问区间内取出两个相同颜色的种类数
思路:由于不是在线更新,那么可以进行离线查询,而且当知道了[l,r]的答案,且能在O(1)的条件下得知[l-1,r],[l+1,r],[l,r+1],[l,r-1]的答案,那么就能使用莫队算法了。 大致上,将区间分块,由于n=a+b>=a*b,显然将区间开平方根是最优的,我们先将询问保存,按照块序第一优先,再考虑右端点进行排序。再来,使用cnt[]来记录当前颜色出现的次数,当得到[l,r]后,再考虑加入[l-1,r] ,[l,r+1],对答案\(ans-=cnt[col[l-1]]^2,ans+=(cnt[[col[l-1]]+1)^2 \)如果是缩小区间,那么反之。
/** @Date : 2016-12-07-21.28 * @Author : Lweleth (SoungEarlf@gmail.com) * @Link : https://github.com/ * @Version : */#include<bits/stdc++.h>#define LL long long#define PII pair#define MP(x, y) make_pair((x),(y))#define fi first#define se second#define PB(x) push_back((x))#define MMG(x) memset((x), -1,sizeof(x))#define MMF(x) memset((x),0,sizeof(x))#define MMI(x) memset((x), INF, sizeof(x))using namespace std;const int INF = 0x3f3f3f3f;const int N = 1e5+20;const double eps = 1e-8;LL col[N];int blc[N];int cnt[N];struct yuu{ LL l, r; int id; LL a, b;}q[N];int cmp(yuu a, yuu b){ if(blc[a.l] == blc[b.l]) return a.r < b.r; return a.l < b.l;}int cmpi(yuu a, yuu b){ return a.id < b.id;}int main(){ int n, m; scanf("%d%d", &n, &m); //{ MMF(cnt); for(int i = 1; i <= n; i++) scanf("%lld", col + i); int ct = sqrt(n); for(int i = 1; i <= n; i++)//分块 blc[i] = (i - 1)/ct + 1; for(int i = 1; i <= m; i++) { scanf("%lld%lld", &q[i].l, &q[i].r); q[i].id = i; } sort(q + 1, q + 1 + m, cmp); LL ans = 0; LL ll = 1, rr = 0; for(int i = 1; i <= m; i++) { //cout << q[i].l << q[i].r << endl; if(ll > q[i].l) for(int j = ll; j > q[i].l; j--) ans += (2*cnt[col[j - 1]] + 1), cnt[col[j - 1]]++; if(ll < q[i].l) for(int j = ll; j < q[i].l; j++) ans -= (2*cnt[col[j]] - 1), cnt[col[j]]--; if(rr < q[i].r) for(int j = rr; j < q[i].r; j++) ans += (2*cnt[col[j + 1]] + 1), cnt[col[j + 1]]++; if(rr > q[i].r) for(int j = rr; j > q[i].r; j--) ans -= (2*cnt[col[j]] - 1), cnt[col[j]]--; ll = q[i].l; rr = q[i].r; if(q[i].l == q[i].r) { q[i].a = 0; q[i].b = 1; continue; } q[i].b = (q[i].r - q[i].l) * (q[i].r - q[i].l + 1); q[i].a = ans - (q[i].r - q[i].l + 1); LL g = __gcd(q[i].b, q[i].a); //cout << ans <<endl; q[i].a /= g; q[i].b /= g; } sort(q + 1, q + 1 + m, cmpi); for(int i = 1; i <= m; i++) printf("%lld/%lld\n", q[i].a, q[i].b); //} return 0;}
bzoj 2038 莫队入门
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。