码迷,mamicode.com
首页 > 其他好文 > 详细

HDU-4455 Substrings(DP)

时间:2016-07-31 19:00:25      阅读:165      评论:0      收藏:0      [点我收藏+]

标签:

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

  

HDU-4455 Substrings(DP)

标签:

原文地址:http://www.cnblogs.com/20143605--pcx/p/5723444.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!