【莫队算法】[HDU5213]Lucky

题目

分析：重点是把每个两个区间的询问拆分成四个单个区间的询问。
ans[l][r][u][v]=ans[l][v]-ans[l][u-1]-ans[r+1][v]+ans[r+1][u-1];

#include<cstdio>
#include<map>
#include<cmath>
#include<cstring>
#include<cstdlib>
#include<algorithm>
using namespace std;
#define MAXN 30000
int n,k,m,a[MAXN+10],cm,cnt[MAXN+10],block[MAXN+10],ans[MAXN+10];
struct node{
    int l,r,i,sign;
    bool operator<(const node &x)const{
        if(block[l]==block[x.l])
            return r<x.r;
        return block[l]<block[x.l];
    }
}qi[MAXN*4+10];
struct node2{
    int l,r,u,v,pos,ans;
}qd[MAXN+10];
void Read(int &x){
    char c;
    while(c=getchar(),c!=EOF)
        if(c>='0'&&c<='9'){
            x=c-'0';
            while(c=getchar(),c>='0'&&c<='9')
                x=x*10+c-'0';
            ungetc(c,stdin);
            return;
        }
    exit(0);
}
void init(){
    Read(n),Read(k);
    int i,t=sqrt(n+0.5),l,r,u,v;
    for(i=1;i<=n;i++){
        block[i]=(i+t-1)/t;
        Read(a[i]);
    }
    Read(m);
    for(i=1;i<=m;i++){
        Read(l),Read(r),Read(u),Read(v);
        qi[++cm].l=l;
        qi[cm].r=v;
        qi[cm].i=i,qi[cm].sign=1;
        qi[++cm].l=l;
        qi[cm].r=u-1;
        qi[cm].i=i,qi[cm].sign=-1;
        qi[++cm].l=r+1;
        qi[cm].r=v;
        qi[cm].i=i,qi[cm].sign=-1;
        qi[++cm].l=r+1;
        qi[cm].r=u-1;
        qi[cm].i=i,qi[cm].sign=1;
    }
    sort(qi+1,qi+cm+1);
}
void solve(){
    int i,l=1,r=0,nans=0;
    node *p;
    for(i=1;i<=cm;i++){
        p=&qi[i];
        if(r<p->r)
            while(r<p->r){
                nans+=cnt[k-a[++r]];
                cnt[a[r]]++;
            }
        else
            while(r>p->r){
                cnt[a[r]]--;
                nans-=cnt[k-a[r--]];
            }
        if(l>p->l)
            while(l>p->l){
                nans+=cnt[k-a[--l]];
                cnt[a[l]]++;
            }
        else
            while(l<p->l){
                cnt[a[l]]--;
                nans-=cnt[k-a[l++]];
            }
        ans[p->i]+=p->sign*nans;
    }
}
void print(){
    int i;
    for(i=1;i<=m;i++)
        printf("%d\n",ans[i]);
}
int main()
{
    while(1){
        cm=0;
        memset(cnt,0,sizeof cnt);
        memset(ans,0,sizeof ans);
        init();
        solve();
        print();
    }
}