「USACO 2021 US Open Platinum」United Cows of Farmer John

Posted 2021-05-13 chasedeath

「USACO 2021 US Open Platinum」United Cows of Farmer John

考虑依次枚举右端点\\(i\\)，计算左边合法的方案数，设一个数\\(x\\)上次出现的位置为\\(lst_x\\)

则\\(i\\)能够作为右端点的区间就是\\([lst_{a_i}+1,i-2]\\)

考虑什么样的位置可以作为左端点，显然这个点在\\([1,i]\\)中是最后一次出现

我们将不妨这样的点权值设为\\(w_i=1\\)

考虑一个点作为中间点贡献怎样的区间，同样的，这个点在\\([1,i]\\)中是最后一次出现

并且，能够贡献的区间\\(>\\)上一次出现的位置\\(lst_x\\)

这个中间点能够匹配的左端点个数就是\\(\\displaystyle \\sum_{k=lst_{a_j}+1}^{j-1} w_k\\)

现在我们要用数据结构动态修改某一个位置的\\(w_i\\)，增减\\([lst_{a_j}+1,j-1]\\)的区间，查询\\([lst_{a_i}+1,i-2]\\)

不妨再为一个点增加点权\\(t_i\\)，此时我们要维护的操作

1.单点修改\\(w_i\\)

2.区间修改\\(t_i\\)

3.求\\(w_it_i\\)区间和

在线段树上每个节点维护\\(w_i\\)之和，\\(w_it_i\\)之和，可以标记永久化\\(t_i\\)

具体实现参考代码（实际写得很丑）

const int N=2e5+10,INF=1e9+10;


int n;
int lst[N],lst2[N],cnt;
ll s1[N<<2],s2[N<<2];
int t[N<<2];
// s1表示w之和，s2表示区间内部t[i]*w[i]之和，t[i]现在是永久化的标记
void Up(int p){
	s2[p]=s2[p<<1]+s2[p<<1|1];
	s1[p]=s1[p<<1]+s1[p<<1|1]+s2[p]*t[p];
}
void Upd(int p,int l,int r,int x){
	if(l==r) {
		s2[p]^=1,s1[p]=t[p]*s2[p];
		return;
	}
	int mid=(l+r)>>1;
	x<=mid?Upd(p<<1,l,mid,x):Upd(p<<1|1,mid+1,r,x);
	Up(p);
}

void Upd(int p,int l,int r,int ql,int qr,int x){
	if(ql>qr) return;
	if(ql<=l && r<=qr) {
		t[p]+=x,s1[p]+=x*s2[p];
		return;
	}
	int mid=(l+r)>>1;
	if(ql<=mid) Upd(p<<1,l,mid,ql,qr,x);
	if(qr>mid) Upd(p<<1|1,mid+1,r,ql,qr,x);
	Up(p);
}

struct Node{
	ll x,y;
	Node(ll x=0,ll y=0):x(x),y(y){  }
	Node operator + (const Node __) { return Node(x+__.x,y+__.y); }
};
Node Que(int p,int l,int r,int ql,int qr){
	if(ql>qr) return Node();
	if(ql<=l && r<=qr) return Node(s1[p],s2[p]);
	int mid=(l+r)>>1; Node res;
	if(ql<=mid) res=res+Que(p<<1,l,mid,ql,qr);
	if(qr>mid) res=res+Que(p<<1|1,mid+1,r,ql,qr);
	res.x+=res.y*t[p];
	return res;
}

int main(){
	n=rd();
	ll ans=0;
	rep(i,1,n) {
		int x=rd();
		if(lst[x]) {
			Upd(1,1,n,lst[x]),cnt--;
			Upd(1,1,n,lst2[x]+1,lst[x],-1);
		}
		Node t=Que(1,1,n,lst[x]+1,i-2);
		ans+=t.x;
		Upd(1,1,n,i),cnt++,Upd(1,1,n,lst[x]+1,i-1,1);
		lst2[x]=lst[x],lst[x]=i;
	}
	printf("%lld\\n",ans);
}