Educational Codeforces Round 95 (Rated for Div. 2) G. Three Occurrences（永久化懒标线段树 / hash）

题目链接：Problem - G - Codeforces 

 解法1：永久化懒标线段树

参考自：CodeForces - 1418G Three Occurrences(线段树-标记永久化)_Frozen_Guardian的博客-CSDN博客

永久化懒标有些要特别注意的地方，具体看代码注释。

#include <bits/stdc++.h>
using namespace std;
#define FOR(i,a,b) for(int i=(a), (i##i)=(b); i<=(i##i); ++i)
#define ROF(i,a,b) for(int i=(a), (i##i)=(b); i>=(i##i); --i)
template<class T>inline bool cmax(T&a,const T&b){return a<b?a=b,1:0;}
template<class T>inline bool cmin(T&a,const T&b){return a>b?a=b,1:0;}
#define int long long
#define mid (l+r>>1)
#define ls o<<1
#define rs o<<1|1
const int N = 5e5+5;
int n,a[N],t[N<<2],lz[N<<2]; //t[o]记录o区间的和，lz[o]记录o区间的锁的数量
int las[N][5], ans = 0;

void pushup(int o,int len){
	if(lz[o]) t[o]=0;
	else if(len==1) t[o]=1;
	else t[o]=t[ls]+t[rs];
}
void build(int o,int l,int r){
	if(l==r) {t[o]=1; lz[o]=0; return;}
	build(ls,l,mid); build(rs,mid+1,r);
	pushup(o,r-l+1);
}
void upd(int o,int l,int r,int x,int y,int val){
	if(x>y) return;
	if(x<=l && r<=y){
		lz[o]+=val; pushup(o,r-l+1); return;
	}
	if(x<=mid) upd(ls,l,mid,x,y,val);
	if(y >mid) upd(rs,mid+1,r,x,y,val);
	pushup(o,r-l+1);
}
int query(int o,int l,int r,int x,int y){
	//注意，因为懒标不下传，lz不会导致下面t值为0，所以，如果上层有lz的情况下继续下潜，将得到错误答案！
	//所以，这里必须要判lz[o]>0，如果是>0的话直接返回0，不再下潜
	if(x>y || lz[o]>0) return 0;
	if(x<=l && r<=y) return t[o];
	int res=0;
	if(x<=mid) res+=query(ls,l,mid,x,y);
	if(y >mid) res+=query(rs,mid+1,r,x,y);
	return res;
}
signed main(){
    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
	cin>>n; FOR(i,1,n) cin>>a[i];
	build(1,1,n);
	int L=1;
	FOR(i,1,n){
		int x=a[i];
		for(int j=3; j>0; j--) las[x][j]=las[x][j-1]; //之前的x位置
		las[x][0]=i; //当前位置
		upd(1,1,n,las[x][1]+1,las[x][0],1); //上锁
		upd(1,1,n,las[x][3]+1,las[x][2],-1); //开锁（如果上过锁的话）
		cmax(L, las[x][3]+1); //更新有效左区间
		ans += query(1,1,n,L,i);
	}
	cout<<ans;
}

 解法2：对%3的cnt桶进行hash判断，hash值相同答案累加（不好解释，具体看代码）

#include <bits/stdc++.h>
using namespace std;
#define FOR(i,a,b) for(int i=(a), (i##i)=(b); i<=(i##i); ++i)
#define ROF(i,a,b) for(int i=(a), (i##i)=(b); i>=(i##i); --i)
template<class T>inline bool cmax(T&a,const T&b){return a<b?a=b,1:0;}
template<class T>inline bool cmin(T&a,const T&b){return a>b?a=b,1:0;}
#define int long long
#define puu pair<ull,ull>
#define ull unsigned long long
const int N = 5e5+5, bs=131;
int n,a[N],cnt[N],ans=0;
ull p[N], h[N];
map<ull,int> mp;
queue<int> q[N];

void solve(){
    cin>>n; p[0]=1; mp[0]=1;
    FOR(i,1,n) p[i]=p[i-1]*bs;

    int L=0; //有效区间左边界
    FOR(i,1,n){
        int x; cin>>x;

        if(q[x].size() >= 4){
            int del=q[x].front(); q[x].pop();
            while(L<=del){
                mp[h[L]]--;
                L++;
            }
        }

        h[i]=h[i-1]; //继承前面的hash值
        h[i]-=cnt[x]*p[x];
        cnt[x]=(cnt[x]+1)%3;
        h[i]+=cnt[x]*p[x];

        ans+=mp[h[i]]; //累计答案
        mp[h[i]]++; //桶hash数值+1
        q[x].push(i); //加入一个位置
    }
    cout<<ans<<'\n';
}
signed main(){
    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    int T=1; //cin>>T;
    while(T--) solve();
}