题目大意:给一个长度为n的整数序列,定义egg(i,j)表示区间[i,j]中不同的数的个数。q次询问,每次询问x,表示求所有长度为x连续区间的 egg 之和。
题目分析:定义dp(len)表示所有长度为len的连续区间的egg之和。
则,dp(len)=dp(len-1)-egg(最后一个长度为len-1的连续区间)+每个长度为len的区间中的新增元素不重复的区间个数。
代码如下:
# include<bits/stdc++.h>
using namespace std;
# define LL long long
# define mid (l+(r-l)/2) const int N=1000000; int a[N+5];
int A[N+5];
int f[N+5];
int cnt[N+5];
LL dp[N+5]; int main()
{
int n;
while(scanf("%d",&n)&&n)
{
memset(f,0,sizeof(f));
memset(cnt,0,sizeof(cnt));
for(int i=1;i<=n;++i){
scanf("%d",a+i);
++cnt[i-f[a[i]]];
f[a[i]]=i;
}
memset(f,0,sizeof(f));
A[n+1]=0;
for(int i=n;i>=1;--i){
if(f[a[i]]) A[i]=A[i+1];
else A[i]=A[i+1]+1;
f[a[i]]=1;
}
dp[1]=n;
int sum=n;
for(int i=2;i<=n;++i){
dp[i]=dp[i-1]-A[n-i+2];
sum-=cnt[i-1];
dp[i]+=sum;
}
int m,x;
scanf("%d",&m);
while(m--)
{
scanf("%d",&x);
printf("%lld\n",dp[x]);
}
}
return 0;
}