NOI Online #2 提高组 第二题：子序列问题

前言

我这道题考试时候也爆零了。

但是，注意，这个一百分题解很长那是因为很详细

题目

传送门

解析

暴力

最暴力的做法即为枚举每一对$l$，$r$，并分别计算其 $f(l,r)$ 的值。但这样不知道会慢到哪里去了，所以我们考虑只滚动$l$的值，并每次计算 $G(l)={\sum }_{r=l}^n{f}^2(l,r)$ 的值。

转移

考虑 $G(l)$ 如何从 $G(l-1)$ 转移过来，由 $G$ 的定义：

$$G(l-1)=f^2(l-1,l-1)+f^2(l-1,l)\cdots +f^2(l-1,n)$$
$$G(l)=f^2(l,l)+\cdots +f^2(l,n)$$

不难发现，$G(l)$与$G(l-1)$的唯一区别在于$G(l)$中的每个 $f(l,i)$ 的值都少考虑了 $A_{l-1}$​，即：

$$\begin{array}{rl}f(l-1,l-1)& \to 0\\ f(l-1,l)& \to f(l,l)\\ f(l-1,l+1)& \to f(l,l+1)\\ f(l-1,l+2)& \to f(l,l+2)\\ & ⋮\end{array}$$

（$f(l-1,l-1)$彻底没了）。

所以为了计算$A_{l-1}$对哪些$f$有影响，我们可以先预处理出一个数组 $suf[i]$，表示 $A_x=A_i$​ 且 $x>i$ 时，$x$ 的最小值（如果找不到这样的数，则 $suf[i]=n+1$）。

在从$G(l-1)$ 转移到$G(l)$的过程中，$A_{l-1}$的消失只对 $f(l-1,l-1\sim suf[l-1]-1)$有影响（会-1）,因为 $f(l-1,suf[l-1]\sim n)$ 中，都有超过两个与$A_{l-1}$相等的元素，所以即使$A_{l-1}$消失了，这些区间内数字的种类数也不会变。

即：
$$f(l,r)=\{\begin{array}{rlrl}& f(l-1,r)+1& & (r\in [l-1,suf[l-1]-1])\\ & f(l-1,r)& & (r\in [suf[l-1],n])\end{array}$$

这样，我们就可以通过线段树维护$f(l-1,i)$，在 $\log (n)$的时间内，将每一个$f(l-1,i)$ 转移成$f(l,i)$（区间修改）。

但是这题答案要求的是$f(l,i)$的平方和，怎么办？

由前缀和维护平方和

对于$f(l-1,r)$ ,如果$f(l,r)=f(l-1,r)-1$那么

$$\begin{array}{rl}{f}^2(l,r)& =(f(l-1,r)-1{)}^2\\ & =f^2(l-1,r)-(2f(l-1,r)-1)\end{array}$$

又因为

$$\begin{array}{rl}G(l-1)-G(l)=(f^2(l-1,l-1)& -0)+\\ (f^2(l-1,l)& -f^2(l,l))+\\ (f^2(l-1,l+1)& -f^2(l,l+1))+\\ & ⋮\\ (f^2(l-1,n)& -f^2(l,n))\end{array}$$

又因为刚才推出$r\in [suf[l-1],n]$时，$f(l-1,r)$ 与 $f(l,r)$ 相同，所以式子中许多项被消掉了（式子中的 $n$ 变成了 $suf[l-1]$）

$$\begin{array}{rl}G(l-1)-G(l)=(f^2(l-1,l-1)& -0)+\\ (f^2(l-1,l)& -f^2(l,l))+\\ (f^2(l-1,l+1)& -f^2(l,l+1))+\\ & ⋮\\ (f^2(l-1,suf[l-1]-1)& -f^2(l,suf[l-1]-1))\end{array}$$

又由于第二部分的结论（$r\in [l-1,suf[l-1]-1]$时，$f(l-1,r)=f(l,r)+1$）和这部分开头的结论，式子可以再次化简：

$$\begin{array}{rlrl}G(l)-G(l-1)& =& -(2f(l-1,l-1)& -1)\\ & & -(2f(l-1,l)& -1)\\ & & -(2f(l-1,l+1)& -1)\\ & & & ⋮\\ & & -(2f(l-1,suf[l-1]-1)& -1)\\ & =& -2\times \sum _{r=l-1}^{n}f(l-1,r)+((suf[l-1])& -(l-1)+1)\end{array}$$

这个值可以直接用刚才维护$f$的线段树来算出。

代码

#include<bits/stdc++.h>
using namespace std;
const long long Mod=1000000007;
const long long MXN=1e6+5;
long long n;
long long arr[MXN],pool[MXN];
long long suf[MXN],nxt[MXN],diff[MXN];
long long tot=0,ans=0;
struct set_tree{ //线段树板子
	long long t[MXN<<2],tag[MXN<<2];
	long long ls(long long p){return p<<1;}
	long long rs(long long p){return (p<<1)|1;}
	void push_up(long long p){t[p]=t[ls(p)]+t[rs(p)];}
	void add_tag(long long p,long long l,long long r,long long k){
		tag[p]+=k;
		t[p]+=k*(r-l+1);
	}
	inline void push_down(long long p,long long l,long long r){
		if(!tag[p])return;
		long long mid=(l+r)>>1;
		add_tag(ls(p),l,mid,tag[p]);
		add_tag(rs(p),mid+1,r,tag[p]);
		tag[p]=0;
	}
	void build(long long p,long long l,long long r){
		tag[p]=0;
		if(l==r){
			t[p]=diff[l];
			return;
		}
		long long mid=(l+r)>>1;
		build(ls(p),l,mid);
		build(rs(p),mid+1,r);
		push_up(p);
	}
	void mo(long long p,long long l,long long r,long long al,long long ar,long long k){
		if(al>ar)return;
		if(al<=l && r<=ar){
			add_tag(p,l,r,k);
			return;
		}
		long long mid=(l+r)>>1;
		push_down(p,l,r);
		if(al<=mid)mo(ls(p),l,mid,al,ar,k);
		if(ar>mid)mo(rs(p),mid+1,r,al,ar,k);
		push_up(p);
	}
	long long query(long long p,long long l,long long r,long long al,long long ar){
		if(al>ar)return 0;
		if(al<=l && r<=ar) return t[p];
		long long mid=(l+r)>>1;
		push_down(p,l,r);
		long long res=0;
		if(al<=mid)res+=query(ls(p),l,mid,al,ar);
		if(ar>mid)res+=query(rs(p),mid+1,r,al,ar);
		return res;
	}
}Tree;
int main()
{
	scanf("%lld",&n);
	for(long long i=1;i<=n;i++)
	{
		scanf("%lld",arr+i);
		pool[i]=arr[i];
		nxt[i]=n+1;
	}
	sort(pool+1,pool+1+n);
	long long cnt=unique(pool+1,pool+1+n)-pool-1;
	for(long long i=1;i<=n;i++)
		arr[i]=lower_bound(pool+1,pool+1+cnt,arr[i])-pool;
	for(long long i=n;i>=1;i--)
	{
		suf[i]=nxt[arr[i]];
		nxt[arr[i]]=i;
	}
	memset(pool,0,sizeof(pool));
	for(long long i=1;i<=n;i++){
		diff[i]=diff[i-1];
		diff[i]+=((pool[arr[i]]++)==0);
		tot+=diff[i]*diff[i];
		tot%=Mod;
	}
	Tree.build(1,1,n);
	for(long long i=1;i<=n;i++)
	{
		ans+=tot;
		ans%=Mod;
		tot-=(Tree.query(1,1,n,i,suf[i]-1)*2)-(suf[i]-i);
		tot%=Mod;
		Tree.mo(1,1,n,i+1,suf[i]-1,-1);
	}
	ans%=Mod;
	if(ans<0)ans+=Mod;
	printf("%lld",ans);
	return 0;
}