【动态规划】聪明伶俐的香穗子

Problem 4 聪明伶俐的香穗子

  香穗子遇到难题了.

  题目是这样的,一个序列上有n个整数,现在你要取m个,且这m个数的任意两个不能相隔的太近,否则这样会太丑,现在问你最大能得到多大的和

 

输入:

       第一行三个数n,m,k,分别表示n个数,取m个,且m个中的任意两个位置差要大于等于K

       接下来一行,有n个整数,表示序列上的每个数

 

输出:

       最大和

 

Sample Input

       4 2 2

       3 4 -5 1

 

Sample Output

       5

 

数据范围:

       n<=10000,m<=100，m<=n

       答案保正小于 Maxlongint

看起来很像是单调队列，但是完全用不着，单调队列怎么说也要难调试一些。

枚举最后一选，考虑到最后一选与上次一选有关系，但是只是上一选的位置有关系，没有其他更多关系，因此，我们不需要最后一选的状态记录。

f[i][j] = max(f[k][j-1] + a[i]),i-k>=K。因此就限定了k的取值区间，但是因为已经说过与最后一选没有关系，因此我们可以转换一下状态表示，表示“前”i个数选j个。

因此f[i][j]=max(f[i-1][j] , f[i-K][j-1]+a[i])。

用单调队列也不是没法做。

f[i][j] = max(f[k][j-1] + a[i]),i-k>=K。

观察发现，只与上一行的j左边的区域有关，因此我们可以让动规和单调队列的维护刚好异步一行的位置，即用上一行的j左边的f值来维护单调队列。

但是要注意每一行对应一个单调队列，所以枚举i的时候一定要将单调队列清空，一开始就因为这个调试了很久。

#include <cstdio>
#include <string>
#define min(a,b) ((a)<(b)?(a):(b))
#define max(a,b) ((a)>(b)?(a):(b))

long a[10010];
long f[10010][101];

long getint()
{
	long rs=0;bool sgn=1;char tmp;
	do tmp = getchar();
	while (!isdigit(tmp)&&tmp-'-');
	if (tmp == '-'){tmp=getchar();sgn=0;}
	do rs=(rs<<3)+(rs<<1)+tmp-'0';
	while (isdigit(tmp=getchar()));
	return sgn?rs:-rs;	
}

int main()
{
	freopen("4.in","r",stdin);
	freopen("4.out","w",stdout);

	long n = getint();
	long m = getint();
	long k = getint();
	for (long i=1;i<n+1;i++)
		a[i] = getint();
	memset(f,~0x3f,sizeof f);
	for (long i=1;i<n+1;i++)
	{
		f[i][1] = max(f[i-1][1],a[i]);
	}
	for (long i=k;i<n+1;i++)
	{
		for (long j=2;j<m+1;j++)
		{
			f[i][j] = max(f[i][j],f[i-1][j]);
			if (i >= k)
				f[i][j] = max(f[i][j],f[i-k][j-1]+a[i]);
		}
	}
	printf("%ld",f[n][m]);

	return 0;
}

#include <cstdio>
#include <string>
#include <cstring>
#define min(a,b) ((a)<(b)?(a):(b))
#define max(a,b) ((a)>(b)?(a):(b))

long getint()
{
	long rs=0;bool sgn=1;char tmp;
	do tmp = getchar();
	while (!isdigit(tmp)&&tmp-'-');
	if (tmp == '-'){tmp=getchar();sgn=0;}
	do rs=(rs<<3)+(rs<<1)+tmp-'0';
	while (isdigit(tmp=getchar()));
	return sgn?rs:-rs;	
}

long a[10010];
long que[10010];
long f[10010][2];

int main()
{
	freopen("4.in","r",stdin);
	freopen("4.out","w",stdout);

	long n = getint();
	long m = getint();
	long k = getint();

	if (m == 0)
	{
		printf("0");
		return 0;
	}

	long l = 0;
	long r = 0;
	long ths = 0;
	long pre = 1;
	for (long i=1;i<n+1;i++)
	{
		a[i] = getint();
		f[i][ths] = a[i];
	}
	f[0][ths] = -0x3f3f3f3f;
	
	for (long j=2;j<m+1;j++)
	{
		ths ^= 1; pre ^= 1;
		for (long i=0;i<n+1;i++)
			f[i][ths] = -0x3f3f3f3f;

		for (long i=k+1;i<n+1;i++)
		{
			if (f[i-k][pre] > -0x3f3f3f3f)
			{
				while (l<r&&que[r]<=f[i-k][pre]) r--;
				que[++r] = f[i-k][pre];
				f[i][ths] = que[l+1] + a[i];
			}
		}
		l = 0;
		r = 0;
	}

	long ans = 0;
	for (long i=0;i<n+1;i++)
	{
		ans = max(ans,f[i][ths]);
	}

	printf("%ld",ans);

	return 0;
}