HDU 5213 莫队+容斥原理

http://acm.hdu.edu.cn/showproblem.php?pid=5213（题目）

题意：给一个序列a，以及K，有Q个询问，每个询问四个数，L,R,U,V, 求L<=i<=R,U<=j<=V，a[i]+a[j]=K的(i, j)对数（题目保证了L <= R < U <= V）。

ans=f（L,V）-f(L,U)-f(R,V)+f(R,U); 利用四次莫队 容斥求ans

#include<bits/stdc++.h>
using namespace std;
const int maxn=30005;
struct node{
    int l1,r1,l2,r2,id;
}s[maxn];
int a[maxn],c[2*maxn],t,n,m,sum[maxn],k;
int cmp(node x,node y) 
{
    if(x.l1/t!=y.l1/t) return x.l1<y.l1;
    return x.r2<y.r2;
}
int cmp1(node x,node y) 
{
    if(x.l1/t!=y.l1/t) return x.l1<y.l1;
    return x.l2<y.l2;
}
int cmp2(node x,node y) 
{
    if((x.r1+1)/t!=(y.r1+1)/t) return x.r1<y.r1;
    return x.r2<y.r2;
}
int cmp3(node x,node y)
{
    if((x.r1+1)/t!=(y.r1+1)/t) return x.r1<y.r1;
    return x.l2<y.l2;
}
int l,r,ans;
int solve(int L,int R)
{
    while(l<L)
    {
        c[a[l]]--;
        if(k>a[l])
        ans-=c[k-a[l]];
        l++;
    }
    while(l>L)
    {
        l--;
        if(k>a[l])
        ans+=c[k-a[l]];
        c[a[l]]++;
    }
    while(r>R)
    {
        c[a[r]]--;
        if(k>a[r])
        ans-=c[k-a[r]];
        r--;
    }
    while(r<R)
    {
        r++;
        if(k>a[r])
        ans+=c[k-a[r]];
        c[a[r]]++;
    }
    return ans;
}
int main()
{
    ///freopen("in.txt","r",stdin);
    while(~scanf("%d",&n))
    {
        scanf("%d",&k);
        for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
        scanf("%d",&m);
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%d%d",&s[i].l1,&s[i].r1,&s[i].l2,&s[i].r2);
            s[i].id=i;
        }
        t=(int)(sqrt(n)+0.5);
        ///
        sort(s+1,s+m+1,cmp);
        memset(c,0,sizeof c);
        l=1,r=0;ans=0;
        for(int i=1;i<=m;i++)
        sum[s[i].id]=solve(s[i].l1,s[i].r2);
        //
        sort(s+1,s+m+1,cmp1);
        memset(c,0,sizeof c);
        l=1,r=0;ans=0;
        for(int i=1;i<=m;i++)
        sum[s[i].id]-=solve(s[i].l1,s[i].l2-1);
        //
        sort(s+1,s+m+1,cmp2);
        memset(c,0,sizeof c);
        l=1,r=0;ans=0;
        for(int i=1;i<=m;i++)
        sum[s[i].id]-=solve(s[i].r1+1,s[i].r2);
        /
        sort(s+1,s+m+1,cmp3);
        memset(c,0,sizeof c);
        l=1,r=0;ans=0;
        for(int i=1;i<=m;i++)
        sum[s[i].id]+=solve(s[i].r1+1,s[i].l2-1);
        
        for(int i=1;i<=m;i++)
        printf("%d\n",sum[i]);
    }
    return 0;
}