CF617E XOR and Favorite Number(莫隊算法)

題意

給你一個大小爲n的序列,然後給你一個數字k,再給出m組詢問,詢問給出一個區間,問這個區間裏面有多少個區間的異或結果爲k。

分析

因爲這題去學了莫隊QAQ
首先記前綴異或和爲 sis_i,區間異或和爲 kk的對數,等價於 sr xor sl1=ks_r~xor~s_{l-1}=k 的對數。如果 rr 確定了,那麼 sl1=sr xor ks_{l-1}=s_r~xor~k,相當於是在區間中,找某個數出現個數,這其實是莫隊的模板題了。
將詢問排序,左端點如果在同一塊,就按右端點從小到大排序,否則按左端點所在的塊排序。假設我們已經求出了 [l,r][l,r] 這個區間,要轉移到 [l1,r][l-1,r] 這個區間,O(1)O(1) 就可以轉移。那麼對於一個塊,最多有 n\sqrt{n} 個左端點,右端點最多遍佈整個區間,也就是 nn,那麼複雜度是 O(nn)O(n\sqrt{n}) 的。

代碼如下

#include <bits/stdc++.h>
#define LL long long
#define N 2000005
using namespace std;
struct node{
	int l, r, i;
}q[N * 2];
LL z = 1, tot, ans[N];
int bl[N], s[N], k, cnt[N], blo;
int cmp(node x, node y){
	return bl[x.l] == bl[y.l]? (x.r < y.r): bl[x.l] < bl[y.l]; 
}
void add(int x){
	tot += cnt[s[x] ^ k];
	cnt[s[x]]++;
}
void del(int x){
	cnt[s[x]]--;
	tot -= cnt[s[x] ^ k];
}
int main(){
	int i, j, n, m, l, r;
	scanf("%d%d%d", &n, &m, &k);
	blo = n / sqrt(m * 2 / 3);
	if(!blo) blo = sqrt(n);
	for(i = 1; i <= n; i++){
		scanf("%d", &j);
		bl[i] = (i - 1) / blo + 1;
		s[i] = s[i - 1] ^ j;
	}
	for(i = 1; i <= m; i++){
		scanf("%d%d", &q[i].l, &q[i].r);
		q[i].i = i; q[i].l--; 
	}
	sort(q + 1, q + m + 1, cmp);
	l = 1, r = 0;
	for(i = 1; i <= m; i++){
		while(l < q[i].l) del(l), l++;
		while(l > q[i].l) l--, add(l);
		while(r < q[i].r) r++, add(r);
		while(r > q[i].r) del(r), r--;
		ans[q[i].i] = tot;
	}
	for(i = 1; i <= m; i++) printf("%lld\n", ans[i]);
	return 0;
}
發佈了85 篇原創文章 · 獲贊 80 · 訪問量 4008
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章