【ybt 入门到进阶模拟赛 Day1 T3】晨练计划

晨练计划

题目链接：ybt 入门到进阶模拟赛 Day1 T3

题目大意

有 n 天，在每一天你可以选择跑步或者休息，跑步会走的距离一个每天不固定的给出的值，并疲劳值加一。
休息就一定要休息到疲劳值为 0，且一天只能减一个点的疲劳值，然后疲劳值不能超过一个值。
你可以选在在疲劳值为 0 的时候休息，那样疲劳值不变。
问你在这 n 天内，保证最后最后一天之后疲劳值为 0 的情况下，你最远可以走多远。

思路

你考虑 DP。
$f_{i,j}$ 为前 $i$ 天，你做完这 $i$ 天的事之后疲劳值是 $j$ 的情况下，你最远可以走多远。
那首先是跑步，就是 $f_{i-1,j-1}+a_i$。
接着是休息，由于休息是一定要直接休息到 $0$，所以就不用枚举 $j$，直接是 $0$，但是要枚举休息前的疲劳值 $k$，转移就是在 $f_{i-j,j}$ 之间取最大值。
然后由于你可以在疲劳值为 $0$ 的时候休息，所以 $f_{i,0}$ 还可以从 $f_{i-1,0}$ 转移。
（我这里还枚举了 $j$，就是从 $f_{i-j,0}$ 转移，但其实是不用的，是一样的）

然后记得最后要疲劳值是 $0$，就输出 $f_{n,0}$ 即可。

代码

#include<cstdio>
#include<iostream>
#include<cstring>

using namespace std;

int n, m;
int d[10001];
int f[10001][501];

int main() {
//	freopen("run.in", "r", stdin);
//	freopen("run.out", "w", stdout);
	
	memset(f, -1, sizeof(f));
	
	scanf("%d %d", &n, &m);
	for (int i = 1; i <= n; i++)
		scanf("%d", &d[i]);
	
	f[0][0] = 0;
	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= min(i, m); j++) {//休息
			if (f[i - j][0] != -1) f[i][0] = max(f[i][0], f[i - j][0]);//从疲劳值为0的时候休息i次
			if (f[i - j][0] != -1) f[i][0] = max(f[i][0], f[i - j][j]);//从疲劳值为i的时候休息i次
		}
		for (int j = 1; j <= m; j++) {
			if (f[i - 1][j - 1] != -1) f[i][j] = max(f[i][j], f[i - 1][j - 1] + d[i]);//跑步
		}
	}
	
	printf("%d", f[n][0]);//最后疲劳值一定是0
	
	return 0;
}