BZOJ5011 [Jx2017]颜色（洛谷P4065）

线段树

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

对每个颜色记录$mx[i]$和$mn[i]$表示$i$在序列中的最大/最小位置。枚举区间右端点$i$，设$j$为满足$max[c[j]]>i$的最接近$j$的位置。可以发现$[\ 1,max[c[j]]\ ]$是不合法的。并且如果$i=mx[c[i]]$，$(\ mn[c[i]],mx[c[i]]\ ]$也是不合法的。那么$i$对答案的贡献就是$(j,i]$中未打标记的点的数量。

标记需要支持区间修改和查询，那么线段树就可以了。而$j$用一个栈维护，当$i\not=mx[c[i]]$时就把放入栈中，否则一直弹就好了。

代码：

#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 300005
#define F inline
using namespace std;
typedef long long LL;
struct tree{ int l,r,x,f; }t[N<<2];
int n,T,tp,mc,stk[N],mx[N],mn[N],c[N];
LL ans;
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(LL x){ if (x>9) writec(x/10); putchar(x%10+48); }
F void _write(LL x){ writec(x),puts(""); }
void build(int x,int l,int r){
    t[x]=(tree){l,r,0,0};
    if (l==r) return; int mid=l+r>>1;
    build(x<<1,l,mid),build(x<<1|1,mid+1,r);
}
F void pshd(int x){
    t[x<<1].f=t[x<<1|1].f=1,t[x].f=0;
    t[x<<1].x=t[x<<1].r-t[x<<1].l+1;
    t[x<<1|1].x=t[x<<1|1].r-t[x<<1|1].l+1;
}
void mdfy(int x,int l,int r){
    if (t[x].l>r||t[x].r<l) return;
    if (t[x].l>=l&&t[x].r<=r){
        t[x].f=1,t[x].x=t[x].r-t[x].l+1; return;
    }
    if (t[x].f) pshd(x);
    mdfy(x<<1,l,r),mdfy(x<<1|1,l,r);
    t[x].x=t[x<<1].x+t[x<<1|1].x;
}
int srch(int x,int l,int r){
    if (t[x].l>r||t[x].r<l) return 0;
    if (t[x].l>=l&&t[x].r<=r) return t[x].x;
    if (t[x].f) pshd(x);
    return srch(x<<1,l,r)+srch(x<<1|1,l,r);
}
int main(){
    for (T=_read();T;T--){
        n=_read(),mc=ans=tp=0,build(1,1,n);
        for (int i=1;i<=n;i++) mc=max(mc,c[i]=_read());
        for (int i=1;i<=mc;i++) mn[i]=n,mx[i]=0;
        for (int i=1;i<=n;i++) mn[c[i]]=min(mn[c[i]],i);
        for (int i=1;i<=n;i++) mx[c[i]]=max(mx[c[i]],i);
        for (int i=1;i<=n;i++){
            if (mx[c[i]]==i&&mn[c[i]]!=mx[c[i]])
                mdfy(1,mn[c[i]]+1,mx[c[i]]);
            else stk[++tp]=i;
            for (;tp&&mx[c[stk[tp]]]<=i;tp--);
            if (i!=stk[tp]) ans+=i-stk[tp]-srch(1,stk[tp]+1,i);
        }
        _write(ans);
    }
    return 0;
}