题意:
给你一个长度为n的串,问你有多少个子串,使得这个子串中所有种类字符都正好出现3次
题解:
首先我们考虑对于每一个右端点,都有一个左端点的区间使得它合法。那么我们枚举所有右端点,询问合法的左端点的区间中有多少合法的位置。
那么首先考虑线段树,对于当前枚举到的右端点,首先将它距离上一个这个种类的字符的位置的不合法度+1(f表示不合法的区间个数)。
然后再将从右往左第4个到第3个区间中的位置不合法度-1,并且将1~从右往左第四个这个区间的不合法度+1.由于之后前面的区间绝对不合法,所以不用考虑那么细致。
然后查询合法的区间。
对于f标记,我们不能push_down,因为你在更新的时候,无法更新到更下面的层,所以下面层的数量会出问题,如果一个一个更新的话,会T。那么久考虑标记永久化。直接对于当前合法区间操作,这道题有一个特殊的性质,就是加减操作一定会在同一个区间,也就是说对于加操作,要么它在之后一定都不能被取,要么之后一定会有相同区间的一个减操作将他的影响消掉。
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=5e5+5;
int num[N*4],f[N*4],can[N*4];
void build(int l,int r,int root){
if(l==r){
num[root]=1;
return ;
}
int mid=l+r>>1;
build(l,mid,root<<1);
build(mid+1,r,root<<1|1);
num[root]=num[root<<1]+num[root<<1|1];
}
void update(int l,int r,int root,int ql,int qr,int v){
if(l>=ql&&r<=qr){
f[root]+=v;
if(f[root])num[root]=0;
else if(l==r)num[root]=1;
else num[root]=num[root<<1]+num[root<<1|1];
return ;
}
int mid=l+r>>1;
if(mid>=ql)
update(l,mid,root<<1,ql,qr,v);
if(mid<qr)
update(mid+1,r,root<<1|1,ql,qr,v);
if(f[root])num[root]=0;
else if(l==r)num[root]=1;
else num[root]=num[root<<1]+num[root<<1|1];
}
int query(int l,int r,int root,int ql,int qr){
if(f[root])return 0;
if(l>=ql&&r<=qr)return num[root];
int mid=l+r>>1;
int ans=0;
if(mid>=ql)
ans=query(l,mid,root<<1,ql,qr);
if(mid<qr)
ans+=query(mid+1,r,root<<1|1,ql,qr);
return ans;
}
deque<int>pos[N];
int a[N];
int main()
{
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++)scanf("%d",&a[i]);
ll ans=0;
build(1,n,1);
for(int i=1;i<=n;i++){
if(!pos[a[i]].size())
update(1,n,1,1,i,1);
else
update(1,n,1,pos[a[i]].back()+1,i,1);
pos[a[i]].push_back(i);
if(pos[a[i]].size()==4){
int p=pos[a[i]].front();
pos[a[i]].pop_front();
update(1,n,1,p+1,pos[a[i]].front(),-1);
update(1,n,1,1,p,1);
ans+=query(1,n,1,p+1,pos[a[i]].front());
}
else if(pos[a[i]].size()==3){
update(1,n,1,1,pos[a[i]].front(),-1);
ans+=query(1,n,1,1,pos[a[i]].front());
}
}
printf("%lld\n",ans);
return 0;
}