关闭

hdu3113 Lucky 莫队算法

标签: ACM
214人阅读 评论(0) 收藏 举报
分类:
定义记号f(A,B)表示询问区间A,B时的答案
用记号+表示集合的并
利用莫队算法我们可以计算出任意f(A,A)的值
不妨假设A=[l1,r1],B=[l2,r2],C=[r1+1,l21]
容易知道f(A,B)=f(A+B+C,A+B+C)+f(C,C)f(A+C,A+C)f(C+B,C+B)
因此一个询问被拆成四个可以用莫队算法做的询问
直接把上述四个询问拆开即可。
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
const int maxn=120005;
int a[maxn];
int cnt[maxn];
typedef long long LL;
LL s1[maxn];
struct pi{
    int l,r,x,id;
}pp[maxn];
int tot;
void  add(int l,int r,int p,int id){
    pp[tot].l=l;
    pp[tot].r=r;
    pp[tot].x=(l-1)/p+1;
    pp[tot].id=id;
    tot++;
}
int cmp(pi a,pi b){
    if(a.x!=b.x) return a.x<b.x;
    return a.r<b.r;
}
int main()
{
    int i,j,n,m,k;
    while(cin>>n){
        memset(cnt,0,sizeof(cnt));
        cin>>k;
        for(i=1;i<=n;i++) scanf("%d",&a[i]);
        LL s=0;
        cin>>m;
        tot=0;
        int p=sqrt(n);
        for(i=0;i<m;i++){
            int a,b,c,d;
            scanf("%d%d%d%d",&a,&b,&c,&d);
            add(a,d,p,4*i);
            add(b+1,c-1,p,4*i+1);
            add(a,c-1,p,4*i+2);
            add(b+1,d,p,4*i+3);
        }
        sort(pp,pp+tot,cmp);
        int l=0,r=0;
        for(i=0;i<tot;i++){
            if(i==0){
                for(j=pp[i].l;j<=pp[i].r;j++){
                    cnt[a[j]]++;
                    if(a[j]<=k&&k-a[j]<=30000){
                        int p=cnt[k-a[j]];
                        if(p>0) s+=p;
                    }
                }
            }
            else{
                if(l<pp[i].l){
                    for(j=l;j<pp[i].l;j++){
                        if(a[j]<=k&&k-a[j]<=30000){
                            int p=cnt[k-a[j]];
                            if(p>0) s-=p;
                        }
                        cnt[a[j]]--;
                    }
                }
                else{
                    for(j=l-1;j>=pp[i].l;j--){
                        cnt[a[j]]++;
                        if(a[j]<=k&&k-a[j]<=30000){
                            int p=cnt[k-a[j]];
                            if(p>0) s+=p;
                        }
                    }
                }
                if(r<pp[i].r){
                    for(j=r+1;j<=pp[i].r;j++){
                        cnt[a[j]]++;
                        if(a[j]<=k&&k-a[j]<=30000){
                            int p=cnt[k-a[j]];
                            if(p>0) s+=p;
                        }
                    }
                }
                else{
                    for(j=r;j>pp[i].r;j--){
                        if(a[j]<=k&&k-a[j]<=30000){
                            int p=cnt[k-a[j]];
                            if(p>0) s-=p;
                        }
                        cnt[a[j]]--;
                    }
                }
            }
            s1[pp[i].id]=s;
            l=pp[i].l;
            r=pp[i].r;
        }
        for(i=0;i<m;i++){
            LL ss=s1[4*i]+s1[4*i+1]-s1[4*i+2]-s1[4*i+3];
            printf("%I64d\n",ss);
        }
    }
}


0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:87244次
    • 积分:3100
    • 等级:
    • 排名:第11544名
    • 原创:230篇
    • 转载:0篇
    • 译文:0篇
    • 评论:14条