显然,任意一种合法的删颜色的方式都和一对合法的切环的断点一一对应。
然后就和BZOJ4382的第一问一样了。
//反正我就把一个
≤
改成了
=
<script type="math/tex" id="MathJax-Element-4">=</script>就过了
代码:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<queue>
#include<cmath>
using namespace std;
const int maxn=300010;
int n,c[maxn],pre[maxn],lst[maxn];
vector<int> a[maxn];
int read()
{
int x=0;char ch=getchar();
for(;ch<'0'||ch>'9';ch=getchar());
for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
return x;
}
struct heap
{
priority_queue<int> Q,D;
void init()
{
while(!Q.empty()) Q.pop();
while(!D.empty()) D.pop();
}
void push(int x) {Q.push(x);}
void erase(int x) {D.push(x);}
int top()
{
while(!D.empty()&&Q.top()==D.top()) Q.pop(),D.pop();
return Q.empty()?0:Q.top();
}
}H;
struct tree
{
int l,r,sum;
bool b,cvr;
tree *ls,*rs;
tree(){ls=rs=NULL;cvr=0;}
void cal(){sum=0;b=0;}
void update(){b=ls->b||rs->b; sum=ls->sum+rs->sum;}
void pushdown()
{
if(cvr) ls->cvr=rs->cvr=1,ls->cal(),rs->cal(),cvr=0;
}
void build(int lx,int rx)
{
l=lx;r=rx;b=1;sum=r-l+1;
if(l==r) return;
int mid=(l+r)>>1;
(ls=new tree)->build(lx,mid);
(rs=new tree)->build(mid+1,rx);
}
void del(int lx,int rx)
{
if(lx>rx) return;
if(lx==l&&rx==r) {cvr=1;cal();return;}
pushdown();
int mid=(l+r)>>1;
if(rx<=mid) ls->del(lx,rx);
else if(lx>mid) rs->del(lx,rx);
else ls->del(lx,mid),rs->del(mid+1,rx);
update();
}
int qry(int lx,int rx)
{
if(lx>rx) return 0;
if(l==lx&&r==rx) return sum;
pushdown();
int mid=(l+r)>>1;
if(rx<=mid) return ls->qry(lx,rx);
else if(lx>mid) return rs->qry(lx,rx);
else return ls->qry(lx,mid)+rs->qry(mid+1,rx);
}
}*xtr;
int main()
{
int ca=read();
while(ca--)
{
n=read();
for(int i=1;i<=n;i++)
c[i]=read();
(xtr=new tree)->build(1,n);
for(int i=1;i<=n;i++)
a[c[i]].clear();
for(int i=n;i;i--)
a[c[i]].push_back(i);
H.init();
long long ans1=0;
for(int i=1;i<=n;i++)
{
H.push(i);
if(a[c[i]][0]==i)
{
for(int j=0;j<a[c[i]].size();j++)
H.erase(a[c[i]][j]);
xtr->del(*(a[c[i]].end()-1)+1,a[c[i]][0]);
}
int lft=H.top()+1;
ans1+=xtr->qry(lft,i);
}
printf("%lld\n",ans1);
}
return 0;
}