首页 > 代码库 > bzoj1878--离线+树状数组

bzoj1878--离线+树状数组

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

首先预处理出数组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

 

bzoj1878--离线+树状数组