HDU 5213 Lucky 容斥+莫队

HDU 5213
题意:给出长度为n的序列a,和一个奇数K. 
m次询问[l,r,u,v] l<=r<u<=v.问从[l,r]，[u,v]两个区间中分别选一个数x,y.问满足x+y=k的(x,y)对数
n,m<=3e4. a[i]<=n.

记录两组询问区间中每个数的出现频率 增加一个数时容易知道它的贡献,莫队排序后离线一下.
[l,r,u,v] 每个询问有两个区间 不能直接排序? 所以把这两个不相关的区间拆成4个独立的区间 .f[l,r]为l,r内两数加为k的对数.

res[l,r,u,v] = f[l,v] - f[l,u-1] - f[r+1,v] + f[r+1,u-1]. 

#include <bits/stdc++.h>using namespace std;typedef long long ll;const int N=3e4+5,M=4*N;struct node{    int l,r,id,sig;}q[M];int n,k,Q,res,tot,a[N],pos[N],fre[N],ans[N];void init(){    res=0,tot=0;    memset(fre,0,sizeof(fre));    int block=sqrt(n);    for(int i=1;i<=n;i++)        pos[i]=(i-1)/block+1;}bool cmp(node a,node b){    if(pos[a.l]==pos[b.l])        return a.r<b.r;    return a.l<b.l;}void add(int p){    int x=a[p];    if(x>k)        return;    fre[x]++;    res+=fre[k-x];}void del(int p){    int x=a[p];    if(x>k)        return;    res-=fre[k-x];    fre[x]--;}void solve(){    for(int i=1,l=1,r=0;i<=tot;i++)    {        for(;r<q[i].r;r++)            add(r+1);        for(;l>q[i].l;l--)            add(l-1);        for(;r>q[i].r;r--)            del(r);        for(;l<q[i].l;l++)            del(l);        ans[q[i].id]+=q[i].sig * res;    }}int main(){    while(~scanf("%d%d",&n,&k))    {        for(int i=1;i<=n;i++)            scanf("%d",&a[i]);        init();        scanf("%d",&Q);        int l,r,u,v;        for(int i=1;i<=Q;i++)        {            ans[i]=0;            scanf("%d%d%d%d",&l,&r,&u,&v);            q[++tot].l=l,q[tot].r=v,q[tot].id=i,q[tot].sig=1;            q[++tot].l=l,q[tot].r=u-1,q[tot].id=i,q[tot].sig=-1;            q[++tot].l=r+1,q[tot].r=v,q[tot].id=i,q[tot].sig=-1;            if(r+1<=u-1)                q[++tot].l=r+1,q[tot].r=u-1,q[tot].id=i,q[tot].sig=1;        }        sort(q+1,q+1+tot,cmp);        solve();        for(int i=1;i<=Q;i++)            printf("%d\n",ans[i]);    }    return 0;}