hdu5213 莫队算法

题目链接

hdu5213

解法

把一次查询通过容斥拆分成四个查询，用一个数组记录不同数字出现个数，本以为是我更新顺序写挫，其实是1和2打窜了。。。 样例依然是对的，我服了，怎么写样例永远对。。

#include
#include
#include

using namespace std;

const int SIZE = 30005;
const int QSIZE = SIZE * 4;
const int BLOCK = 346;

int a[SIZE], res[SIZE];
int N, K, cnt[SIZE], M, ans;

struct node {
    int L, R, i, add;
    void _(int _l, int _r, int _i, int _a){
        L = _l; R = _r; i = _i; add = _a;
    }
} q[QSIZE];

bool cmp(const node &x, const node &y) {
    if(x.L/BLOCK != y.L/BLOCK) return x.L/BLOCK < y.L/BLOCK;
    return x.R < y.R;
}

void add(int pos) {
    if(K - a[pos] > 0 && K - a[pos] <= N) ans += cnt[K - a[pos]];
    cnt[a[pos]]++;//虽然顺序很重要但是这个题里面貌似K是奇数所以无所谓？
}

void remove(int pos) {
    cnt[a[pos]]--;
    if(K - a[pos] > 0 && K - a[pos] <= N) ans -= cnt[K - a[pos]];
}

int main() {
    while(~scanf("%d%d",&N,&K)) {  
        for(int i = 1; i <= N; i++) scanf("%d",&a[i]);
        int Q;
        ans = 0;
        memset(cnt, 0, sizeof(cnt));
        memset(res, 0, sizeof(res));
        scanf("%d",&Q);
        for(int i = 0; i < Q; i++) {
            int l1, r1, l2, r2;
            scanf("%d%d%d%d",&l1,&r1,&l2,&r2);
            q[i*4]._(l1, r2, i, 1);
            q[i*4 + 1]._(l1, l2 - 1, i, -1);
            q[i*4 + 2]._(r1 + 1, r2, i, -1);
            q[i*4 + 3]._(r1 + 1, l2 - 1, i, 1); 
            //注意別把1和2打躥。。。。
        }
        M = 4 * Q;
        sort(q, q + M, cmp);
        int l = 1, r = 1; //已計算部分l...r-1
        for(int i = 0; i < M; i++) {
            int L = q[i].L, R = q[i].R, addn = q[i].add;
            //if(L > R) continue;
            while(l < L) remove(l++);
            while(r > R+1) remove((r--) - 1); 
            while(l > L) add((l--) - 1);
            while(r <= R) add(r++);
            res[q[i].i] += ans * addn;
        }
        for(int i = 0; i < Q; i++) printf("%d\n",res[i]);
    }
}