|洛谷|动态规划|P2690 接苹果

https://www.luogu.org/problem/show?pid=2690

比较简单的DP，但我还是交了3次才AC，能力仍然不足啊

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
#define ms(i,j) memset(i,j, sizeof i);
/*	NOIP2016倒计时，RP++ 
	Luogu: P2690 接苹果
	设f[i][j]为前i个苹果移动j次的最大摘取苹果数
*/
using namespace std; 
int f[1005][35];
int T,K; 
int a;
int main()
{
	scanf("%d%d", &T, &K);
	scanf("%d", &a);
	ms(f,0);
	for (int i=0;i<=K;i++)
	if (a==1)
		if (i%2) f[1][i] = 0; else f[1][i] = 1;//运动了i次，如果是奇数次，那么接不到这个苹果	
	else 
	if (a==2)
		if (i%2) f[1][i] = 1; else f[1][i] = 0;
	for (int i=2;i<=T;i++)
	{
		scanf("%d", &a);
		for (int j=0;j<=K;j++)
		{
			for (int k=0;k<=j;k++)
			{
				int add = 0;
				if (k==0||!(k%2))
				{	if (a==1) add = 1;}
				else
				{	if (a==2) add = 1;}
				f[i][j] = max(f[i][j], f[i-1][k]+add);
				if (k-1>=0) f[i][j] = max(f[i][j], f[i-1][k-1]+1);
			}
		}
	}
	printf("%d\n", f[T][K]);
    return 0;
}