HDU 5213 Lucky（莫队+容斥）

题意：给你n个数和一个k，现在q次询问，每次给你两个区间L~R, U~V, 问你从L~R中取一个x，从U~V中取一个y，使

得x+y = k的方案数。 N(1≤N≤30000)  M(1≤M≤30000)  Li,Ri,Ui,Vi(1≤Li≤Ri<Ui≤Vi≤N)

思路：这题的每次询问有两个区间，不好操作，因为这题的方案数区间与区间之间具有可加性，也就是说询问L~R和

U~V，我们可以求出四个区间，再通过容斥求的需要的，可以变成（L~V）- （L~U-1）- (R+1~ V) + (R+1 ~ U-1)，给

这四个区间加上个标记是加还是减就行了。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn = 1e6+5;
int a[maxn], ans[maxn], num[maxn], n, m, k, cnt, unit, tmp;
struct node
{
    int id, l, r, sta, blk;
    bool operator < (const node &a) const
    {
        if(blk == a.blk) return r < a.r;
        else return blk < a.blk;
    }
}op[maxn];

void add(int x)
{
    if(k-x >= 0) tmp += num[k-x];
    num[x]++;
}

void del(int x)
{
    if(k-x >= 0) tmp -= num[k-x];
    num[x]--;
}

void solve()
{
    memset(ans, 0, sizeof(ans));
    memset(num, 0, sizeof(num));
    int l = 1, r = 0;
    tmp = 0;
    for(int i = 0; i < cnt; i++)
    {
        while(r < op[i].r) add(a[++r]);
        while(r > op[i].r) del(a[r--]);
        while(l < op[i].l) del(a[l++]);
        while(l > op[i].l) add(a[--l]);
        ans[op[i].id] += (op[i].sta) ? tmp : -tmp;
    }
    for(int i = 1; i <= m; i++)
        printf("%d\n", ans[i]);
}

int main(void)
{
    while(cin >> n >> k)
    {
        unit = sqrt(n);
        for(int i = 1; i <= n; i++)
            scanf("%d", &a[i]);
        scanf("%d", &m);
        cnt = 0;
        for(int i = 1; i <= m; i++)
        {
            int l, r, u, v;
            scanf("%d%d%d%d", &l, &r, &u, &v);
            op[cnt++] = node{i, l, v, 1, l/unit};
            op[cnt++] = node{i, l, u-1, 0, l/unit};
            op[cnt++] = node{i, r+1, v, 0, (r+1)/unit};
            op[cnt++] = node{i, r+1, u-1, 1, (r+1)/unit};
        }
        sort(op, op+cnt);
        solve();
    }
    return 0;
}