bzoj1878 [ SDOI2009 ] --离线+树状数组

这题在线做很麻烦,所以我们选择离线。

首先预处理出数组next[i]表示i这个位置的颜色下一次出现的位置。

然后对与每种颜色第一次出现的位置x,将a[x]++。

将每个询问按左端点排序,再从左往右扫,将next[i]++,如果是询问就先返回sum[r]-sum[l-1](sum是a的前缀和)。其中前缀和可以用树状数组维护。

因为做一个询问之前l到r的所有颜色都加且仅加了1,所以答案就是sum[r]-sum[l-1]。

代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
inline char nc(){
    static char buf[100000],*p1=buf,*p2=buf;
    if(p1==p2){
        p2=(p1=buf)+fread(buf,1,100000,stdin);
        if(p1==p2)return EOF;
    }
    return *p1++;
}
inline void read(int& x){
    char c=nc(),b=1;
    for(;!(c>='0'&&c<='9');c=nc())if(c=='-')b=-1;
    for(;c>='0'&&c<='9';x=x*10+c-48,c=nc());x*=b;
}
struct ask{
    int s,e,f;
}b[200001];
inline bool cmp(ask x,ask y){
    return x.s<y.s;
}
int ans[200001],i,j,k=1,n,a[50001],last[1000001],m,c[50001],x,y,next[50001];
inline int lowbit(int x){
    return x&(-x);
}
inline void update(int x){
    for(int i=x;i<=n;i+=lowbit(i))c[i]++;
}
inline int query(int x){
    int ans=0;
    for(int i=x;i>0;i-=lowbit(i))ans+=c[i];
    return ans;
}
int main()
{
    read(n);
    for(i=1;i<=n;i++)read(a[i]);
    read(m);
    for(i=1;i<=m;i++)read(b[i].s),read(b[i].e),b[i].f=i;
    sort(b+1,b+m+1,cmp);
    for(i=1;i<=n;i++)
    if(!last[a[i]]){
        last[a[i]]=i;
        update(i);
    }else{
        next[last[a[i]]]=i;
        last[a[i]]=i;
    }
    for(i=1;i<=m;i++){
        while(k<b[i].s){
            if(next[k])update(next[k]);
            k++;
        }
        ans[b[i].f]=query(b[i].e)-query(b[i].s-1);
    }
    for(i=1;i<=m;i++)printf("%d\n",ans[i]);
    return 0;
}
bzoj1878

 

posted @ 2016-10-01 21:29  gjghfd  阅读(135)  评论(0编辑  收藏  举报