BZOJ2743 [HEOI2012]采花（洛谷P4113）

版权声明：蒟蒻Blog随意转载 https://blog.csdn.net/a1799342217/article/details/82153395

树状数组

BZOJ题目传送门
洛谷题目传送门

妙蛙

对每个位置记一个$nxt$表示这个位置的数下一次出现的位置。刚开始从左往右扫一遍把所有出现两次及以上的数的第一个位置加1。把询问按左端点排序，每次移动左端点的时候撤掉$nxt[l]$的影响，加上$nxt[nxt[l]]$的影响，区间求和即可。

可能讲得不是很清楚，看代码更好理解一点：

#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 2000005
#define F inline
using namespace std;
struct query{ int l,r,id; }q[N];
int n,m,s,a[N],c[N],lst[N],nxt[N],next[N],t[N],ans[N];
F char readc(){
    static char buf[100000],*l=buf,*r=buf;
    if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
    return l==r?EOF:*l++;
}
F int _read(){
    int x=0; char ch=readc();
    while (!isdigit(ch)) ch=readc();
    while (isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=readc();
    return x;
}
F void writec(int x){ if (x>9) writec(x/10); putchar(x%10+48); }
F void _write(int x){ writec(x),puts(""); }
F void mdfy(int x,int w){ for (;x<=n;x+=x&-x) t[x]+=w; }
F int srch(int x){ for (s=0;x;x-=x&-x) s+=t[x]; return s; }
F bool cmp(query a,query b){ return a.l==b.l?a.r>b.r:a.l<b.l; }
int main(){
    n=_read(),_read(),m=_read();
    for (int i=1;i<=n;i++) a[i]=_read();
    for (int i=n;i;i--) nxt[i]=lst[a[i]],lst[a[i]]=i;
    for (int i=1;i<=n;i++) next[i]=nxt[nxt[i]];
    for (int i=1;i<=n;i++) if (++c[a[i]]==2) mdfy(i,1);
    for (int i=1;i<=m;i++) q[i].l=_read(),q[i].r=_read(),q[i].id=i;
    sort(q+1,q+m+1,cmp);
    for (int i=1;i<=m;i++){
        for (int j=q[i-1].l;j<q[i].l;j++){
            if (nxt[j]) mdfy(nxt[j],-1);
            if (next[j]) mdfy(next[j],1);
        }
        ans[q[i].id]=srch(q[i].r)-srch(q[i].l-1);
    }
    for (int i=1;i<=m;i++) _write(ans[i]);
    return 0;
}