Codeforces 703D 树状数组

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/beihai2013/article/details/52138630

Codeforces 703D
题意:
给1e6长度的区间,每个单位代表一个数。
给1e6个询问,每次问区间内出现次数为偶数次的数异或和。
思路:
赛中的时候想莫队水过去,结果pretest都没过233
正解是树状数组。首先假设是奇数次的话,就是一个简单前缀和。现在偶数次,有一个处理技巧就是记录这个区间出现过的数字的异或和。然后就是明显的树状数组统计题。
尝试用线段树写,但是没有成功。
代码:

    #include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <string>
#include <algorithm>
#include <iostream>
#include <map>
using namespace std;
const int MAXN = 1e6 + 5;
int dp[MAXN], n;
int lowbit(int x){return x & -x;}
void add(int u, int v)
{
    while(u <= n) 
        dp[u] ^= v,
        u += lowbit(u);
}
int query(int u)
{
    int res = 0;
    while(u > 0) {
        res ^= dp[u];
        u -= lowbit(u);
    } 
    return res;
}
struct Query
{
    int l, r, id;
    Query(){}
    Query(int _l, int _r, int _id): l(_l), r(_r), id(_id){} 
}que[MAXN];
bool cmp(Query q1, Query q2){return q1.r < q2.r;}
int a[MAXN];
int query(int o, int L, int R, int l, int r)
{
    if(L >= l && R <= r) return dp[o];
    else {
        int res = 0;
        int mid = (L + R) >> 1;
        if(mid >= l) res ^= query(o << 1, L, mid, l, r);
        if(mid < r) res ^= query((o << 1) | 1, mid + 1, R, l, r);
        return res;
    }
}
int ans[MAXN], sum[MAXN];
map<int,int>vis;
int main()
{
    while(scanf("%d", &n) != EOF) {
        for(int i = 1 ; i <= n ; i++) scanf("%d", &a[i]);
        for(int i = 1 ; i <= n ; i++) sum[i] = sum[i - 1] ^ a[i];
        int q; scanf("%d", &q);
        for(int i = 0 ; i < q ; i++) {
            scanf("%d%d", &que[i].l, &que[i].r);
            que[i].id = i;
        }
        vis.clear();
        int r = 0, l = 1;
        sort(que, que + q, cmp);
        for(int i = 0 ; i < q ; i++) {
            while(r < que[i].r) {
                r++;
                add(r, a[r]);
                if(vis[a[r]] == 0) vis[a[r]] = r;
                else {
                    int last = vis[a[r]];
                    add(last, a[r]);
                    vis[a[r]] = r;
                }
            }
            // printf("que = %d %d %d\n", que[i].l, que[i].r, que[i].id);
            ans[que[i].id] = (sum[que[i].r] ^ sum[que[i].l - 1]) ^ query(que[i].r) ^ query(que[i].l - 1);
        }
        for(int i = 0 ; i < q ; i++) printf("%d\n", ans[i]);
    }
    return 0;
}

没有更多推荐了,返回首页