BZOJ 3236: [Ahoi2013]作业

题目

3236: [Ahoi2013]作业

Time Limit: 100 Sec  Memory Limit: 512 MB
Submit: 732  Solved: 271

Description

Input

Output

Sample Input

3 4
1 2 2
1 2 1 3
1 2 1 1
1 3 1 3
2 3 2 3

Sample Output

2 2
1 1
3 2
2 1

HINT

N=100000,M=1000000

Source

By wangyisong1996加强数据

题解

一道卡测评的好题！用传说的莫队算法能水过，就是我们把每个询问x0,y0看作平面上的一个点，那么暴力转移的代价就是两个点之间的距离，我们把x轴分块，这样总的转移的代价和就是O(sqrt(n)*n)的了。由于这道题需要树状数组维护一下，所以这题的复杂度是O(nsqrt(n)logn)的。

代码

/*Author:WNJXYK*/
#include<cstdio>
#include<algorithm>
using namespace std;

const int Maxm=1000000;
struct QuestionNode{
    int l,r;
    int a,b;
    int index;
}; 
QuestionNode quest[Maxm+10];
struct AnsNode{
    int Ans1,Ans2;
};
AnsNode ans[Maxm+10];

const int Maxn=100000;
int cnt[Maxn+10];

int siz;
int n,m;
int num[Maxn+10];

int sum1[Maxn+10],sum2[Maxn+10],cot[Maxn+10]; 

inline int remin(int a,int b){
    if (a<b) return a;
    return b;
}

inline bool cmp(QuestionNode a,QuestionNode b){
    if (cnt[a.l]<cnt[b.l]) return true;
    if (cnt[a.l]==cnt[b.l] && a.r<b.r) return true;
    return false;
}

inline int lowbit(int x){
    return x&-x;
}

inline void add(int c[],int x,int num){
    for (int i=x;i<=Maxn;i+=lowbit(i))c[i]+=num;
}

inline int get(int c[],int x){
    int res=0;
    for (int i=x;i;i-=lowbit(i))res+=c[i];
    return res;
}

int main(){
    scanf("%d%d",&n,&m);
    //GetCnt
    while(siz*siz<n) siz++;
    for (int i=1;i<=siz;i++){
        int st=siz*(i-1)+1,ed=(remin(siz*i,n));
        for (int j=st;j<=ed;j++){
            cnt[j]=i;
        }
    }    
    //GetNum
    for (int i=1;i<=n;i++) scanf("%d",&num[i]);
    //GetQuest
    for (int i=1;i<=m;i++)scanf("%d%d%d%d",&quest[i].l,&quest[i].r,&quest[i].a,&quest[i].b);
    for (int i=1;i<=m;i++)quest[i].index=i;
    //SortQuest
    sort(quest+1,quest+m+1,cmp);
    //GetAns
    int left=1,right=0;
    for (int i=1;i<=m;i++){
        int l=quest[i].l,r=quest[i].r;
        int a=quest[i].a,b=quest[i].b;
        int index=quest[i].index;
        while (left<l){
            cot[num[left]]--;
            add(sum1,num[left],-1);
            if (cot[num[left]]==0) add(sum2,num[left],-1);
            left++;
        }
        while(l<left){
            left--;
            cot[num[left]]++;
            add(sum1,num[left],1);
            if (cot[num[left]]==1) add(sum2,num[left],1);
        }
        while(right<r){
            right++;
            cot[num[right]]++;
            add(sum1,num[right],1);
            if (cot[num[right]]==1) add(sum2,num[right],1);
        }
        while(r<right){
            cot[num[right]]--;
            add(sum1,num[right],-1);
            if (cot[num[right]]==0) add(sum2,num[right],-1);
            right--;
        }
        ans[index].Ans1=get(sum1,b)-get(sum1,a-1);
        ans[index].Ans2=get(sum2,b)-get(sum2,a-1);
    }
    //Print
    for (int i=1;i<=m;i++) printf("%d %d
",ans[i].Ans1,ans[i].Ans2);
    return 0;
}