NOI Online2 提高组 子序列问题 题解

题目传送门

题目大意： 给出一个序列，求 ${\sum }_{1\le i\le j\le n}(f(i,j){)}^2$，其中 $f(i,j)$ 表示 $i$ ~ $j$ 有多少个不同的元素。

题解

思路挺简单的，先考虑固定 $l=1$，然后求出所有 $f(1,r)$ 以及他们的平方和，将 $f(1,r)$ 们放到一个树状数组里面。

然后将 $l$ 一位一位右移，每次右移一位，原来位置的元素就要从 $f(l,r)$ 中剔除掉，设 $next[a_i]$ 为颜色 $a_i$ 下一个出现的位置，那么 $f(l,p)(p\in [l,next[a_i-1]])$ 都要减一，然后看答案的变化：$(x-1{)}^2=x^2-2x+1$，也就是说，答案要减去 $2\times$ $f(l,p)$ 再加 $next[a_i]-i$。

由于这题还有点卡常，所以不要用 $map$（不然只有 $70$ 分），离散化一下即可。

代码如下：

#include <cstdio>
#include <algorithm>
using namespace std;
#define maxn 1000010
#define mod 1000000007
#define ll long long

int n,a[maxn],f[maxn],last[maxn];
int Next[maxn];
ll tr1[maxn],tr2[maxn];
ll change(int x,int y){for(int i=x;i<=n;i+=(i&-i))tr1[i]=(tr1[i]+y)%mod,tr2[i]=(tr2[i]+y*(x-1)+mod)%mod;}
ll sum(int x){ll re=0;for(int i=x;i>=1;i-=(i&-i))re=(re+tr1[i]*x%mod-tr2[i]+mod)%mod;return re;}
ll getsum(int x,int y){return (sum(y)-sum(x-1)+mod)%mod;};
struct disc{int x,y;};
bool compp(disc x,disc y){return x.x<y.x;}
void discretization(int *darr,int dn)
{
	static disc b[maxn];
	for(int i=1;i<=dn;i++)
	b[i].x=darr[i],b[i].y=i;
	sort(b+1,b+dn+1,compp);
	static int tot=0;b[0].x=-9666;
	for(int i=1;i<=dn;i++)
	if(b[i].x!=b[i-1].x)darr[b[i].y]=++tot;
	else darr[b[i].y]=tot;
}
inline char cn()
{
	static char buf[1000010],*p1=buf,*p2=buf;
	return p1==p2&&(p2=(p1=buf)+fread(buf,1,1000000,stdin),p1==p2)?EOF:*p1++;
}
#define cn getchar
void read(int &x)
{
	x=0;int f1=1;char ch=cn();
	while(ch<'0'||ch>'9'){if(ch=='-')f1=-1;ch=cn();}
	while(ch>='0'&&ch<='9')x=x*10+(ch-'0'),ch=cn(); x*=f1;
}

int main()
{
	read(n);
	ll last_ans=0,ans=0;
	for(int i=1;i<=n;i++)read(a[i]);
	discretization(a,n);
	for(int i=1;i<=n;i++)
	{
		f[i]=f[i-1];
		if(!last[a[i]])f[i]++;
		else Next[last[a[i]]]=i;
		last[a[i]]=i;
		last_ans=(last_ans+1ll*f[i]*f[i]%mod)%mod;
	}
	ans=last_ans;
	for(int i=1;i<=n;i++){
		if(!Next[i])Next[i]=n+1;
		change(i,f[i]-f[i-1]);
	}
	for(int i=1;i<=n;i++)
	{
		last_ans=(1ll*last_ans-((2*getsum(i,Next[i]-1)%mod-Next[i]+mod)+i)%mod+mod)%mod;
		ans=(ans+last_ans)%mod;
		change(i,-1);change(Next[i],1);
	}
	printf("%d",ans);
}