(莫隊算法)CodeForces - 617E XOR and Favorite Number

題意:ios

長度爲n的數列,m次詢問,還有一個k。每次詢問詢問詢問從數列的L到R內有多少個連續子序列異或起來等於k。數組

 

分析:spa

由於事先知道這題能夠用莫隊寫,就正好用這題練習莫隊。code

預處理每一個前綴異或和。blog

而後莫隊按分塊排序後,不斷更新,用一個數組cnt[]記錄當前L到R前綴和的數量。排序

R向右拉,新增的數量就是cnt[pre^k],pre表示當前這個R位置的前綴異或和,而後更新一下cnt。string

其餘的也相似。it

算是一個比較好的入門題。io

 

代碼:入門

 1 #include <cstdio>
 2 #include <cstring>
 3 #include <iostream>
 4 #include <cmath>
 5 #include <map>
 6 #include <vector>
 7 #include <algorithm>
 8 
 9 using namespace std;
10 
11 const int maxn = 1100010;
12 const int bk = 400;
13 int n, q, k;
14 struct Query {
15     int l, r, index;
16 };
17 
18 Query query[maxn];
19 
20 bool cmp(Query a, Query b) {
21     if(a.l / bk == b.l / bk)return a.r < b.r;
22     else return a.l / bk < b.l / bk;
23 }
24 
25 long long a[maxn];
26 long long ans[maxn];
27 long long cnt[maxn];
28 long long pre[maxn];
29 long long res = 0;
30 
31 void add(int x) {
32     res += cnt[x ^ k];
33     cnt[x]++;
34 }
35 
36 void del(int x) {
37     cnt[x]--;
38     res -= cnt[x ^ k];
39 }
40 
41 int main() {
42     scanf("%d%d%d", &n, &q, &k);
43     cnt[0] = 1;
44     pre[0] = 0;
45     for(int i = 1; i <= n; i++) {
46         scanf("%lld", &a[i]);
47         pre[i] = pre[i - 1] ^ a[i];
48     }
49     for(int i = 0; i < q; i++) {
50         scanf("%d%d", &query[i].l, &query[i].r);
51         query[i].index = i;
52     }
53     sort(query, query + q, cmp);
54     int left = 1, right = 0;
55     for(int i = 0; i < q; i++) {
56         if(right < query[i].r) {
57             for(int j = right + 1; j <= query[i].r; j++) {
58                 add(pre[j]);
59             }
60         } else {
61             for(int j = right; j > query[i].r; j--) {
62                 del(pre[j]);
63             }
64         }
65         right = query[i].r;
66         if(left < query[i].l) {
67             for(int j = left; j < query[i].l; j++) {
68                 del(pre[j - 1]);
69             }
70 
71         } else {
72             for(int j = left - 1; j >= query[i].l; j--) {
73                 add(pre[j - 1]);
74             }
75         }
76         left = query[i].l;
77         ans[query[i].index] = res;
78     }
79     for(int i = 0; i < q; i++) {
80         printf("%lld\n", ans[i]);
81     }
82 
83 
84     return 0;
85 }
相關文章
相關標籤/搜索