HDU-4455 Substrings（DP）

题目大意：给一个长度为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;
}