[HDU 5213]Lucky：莫队+容斥原理

点击这里查看原题

莫队好题，用F(l1,r1,l2,r2)表示答案，则根据容斥原理，减去两个数均在[r1+1,r2]内的情况，减去两个数均在[l1,l2-1]内的情况，再加上两个数均在[r1+1,l2-1]内的情况即为答案：

F(l1,r1,l2,r2)=f(l1,r2)-f(r1+1,r2)-f(l1,l2-1)+f(r1+1,l2-1)

于是把每个询问分成4个区间进行莫队

/*
User:Small
Language:C++
Problem No.:5213
*/
#include<bits/stdc++.h>
#define ll long long
#define inf 999999999
using namespace std;
const int M=3e4+5;
int n,a[M],pos[M],k,ans[M],cnt[M],res,tot,m;
bool vis[M];
struct no{
    int id,l,r,f;
}p[M<<2];
void init(){
    memset(ans,0,sizeof(ans));
    memset(cnt,0,sizeof(cnt));
    memset(vis,0,sizeof(vis));
    res=tot=0;
}
bool cmp(no a,no b){
    return pos[a.l]!=pos[b.l]?pos[a.l]<pos[b.l]:a.r<b.r;
}
void update(int x){
    if(vis[x]){
        cnt[a[x]]--;
        if(1<=k-a[x]&&k-a[x]<=n) res-=cnt[k-a[x]];
    }
    else{
        cnt[a[x]]++;
        if(1<=k-a[x]&&k-a[x]<=n) res+=cnt[k-a[x]];
    }
    vis[x]^=1;
}
void cal(){
    int l=1,r=0;
    for(int i=1;i<=tot;i++){
        for(int j=r+1;j<=p[i].r;j++) update(j);
        for(int j=r;j>p[i].r;j--) update(j);
        for(int j=l-1;j>=p[i].l;j--) update(j);
        for(int j=l;j<p[i].l;j++) update(j);
        ans[p[i].id]+=res*p[i].f;
        l=p[i].l,r=p[i].r;
    }
}
void solve(){
    init();
    cin>>k;
    int t=sqrt(n);
    for(int i=1;i<=n;i++){
        cin>>a[i];
        pos[i]=(i-1)/t+1;
    }
    cin>>m;
    for(int i=1;i<=m;i++){
        int l1,r1,l2,r2;
        cin>>l1>>r1>>l2>>r2;
        p[++tot]=(no){i,l1,r2,1};
        p[++tot]=(no){i,r1+1,r2,-1};
        p[++tot]=(no){i,l1,l2-1,-1};
        p[++tot]=(no){i,r1+1,l2-1,1};
    }
    sort(p+1,p+tot+1,cmp);
    cal();
    for(int i=1;i<=m;i++)
        cout<<ans[i]<<endl;
}
int main(){
    freopen("data.in","r",stdin);//
    while(cin>>n) solve();
    return 0;
}