bzoj 2743(树状数组)

传送门
题解:对区间有贡献的数是从左端点往后所有第二次出现的数。一开始将每个数第二次出现的位置插入树状数组,然后将区间按x排序从左往右扫,每删除一个点i将nxt[i](i位置的数下一次出现的位置)从树状数组中删除,然后加入nxt[nxt[i]],对于每个区间用右端点在树状数组中的query减去左端点的query即可(不过Po爷好没有做减法也过了,本蒟蒻已发出提问等待“官方”解释。。。)。

#include
using namespace std;
const int MAXN=1e6+2;
int n,waste,m;
int a[MAXN],pre[MAXN],nxt[MAXN],c[MAXN],ans[MAXN];
struct Q {
    int x,y,id;
    friend bool operator <(const Q &a,const Q &b) {
        return a.xx;
    }
}q[MAXN];
bool vis[MAXN];
inline int read() {
    int x=0;char c=getchar();
    while (c<'0'||c>'9') c=getchar();
    while (c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
    return x;
}
inline void modify(int pos,int val) {
    for (int i=pos;i<=n;i+=(i&-i)) c[i]+=val;
}
inline int query(int pos) {
    int ret=0;
    for (int i=pos;i;i-=(i&-i)) ret+=c[i];
    return ret;
}
int main() {
//  freopen("bzoj 2743.in","r",stdin);
    memset(vis,false,sizeof(vis));
    memset(pre,0,sizeof(pre));
    n=read(),waste=read(),m=read();
    for (register int i=1;i<=n;++i) a[i]=read();
    for (register int i=1;i<=n;++i) {
        if (pre[a[i]]) nxt[pre[a[i]]]=i;
        else vis[i]=true;
        pre[a[i]]=i;
    }
    for (register int i=1;i<=n;++i)
        if (vis[i]&&nxt[i]) modify(nxt[i],1);
    for (register int i=1;i<=m;++i) q[i].x=read(),q[i].y=read(),q[i].id=i;
    sort(q+1,q+m+1);
    int now=1;
    for (register int i=1;i<=m;++i) {
        while (now<q[i].x) {
            if (nxt[now]) {
                modify(nxt[now],-1);
                if (nxt[nxt[now]])
                    modify(nxt[nxt[now]],1);
            }
            ++now;
        }
        ans[q[i].id]=query(q[i].y)-query(q[i].x);
    }
    for (register int i=1;i<=m;++i) printf("%d\n",ans[i]);
    return 0;
}

你可能感兴趣的:(数据结构)