思路:动态规划题, 用dp[i][j] 表示奶牛第i秒疲劳值为j时所能跑的最长路程, length[]
存储输入数据, 主要是dp[i][0]计算时较为复杂(转移方程较多), 要考虑原值,
(1~m)分前开始休息到i秒疲劳值刚好为0, 还有就是从(1~m)分前疲劳值为0就开始
休息(根本没跑)。最后dp[n][0]就是所求答案。
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#define MIN -9999999
int dp[2001][501], length[2001];
int mymax(int a, int b, int c) //返回三个数中最大的
{
if(a >= b && a >= c)
{
return a;
}
else if(b >= a && b >= c)
{
return b;
}
else if(c >= a && c >= b)
{
return c;
}
return -1;
}
int main()
{
int n, m, i, j, k, tmax;
while(scanf("%d%d", &n, &m) != EOF)
{
for(i = 1; i <= n; i++)
{
scanf("%d", &length[i]); //接收数据
dp[0][i] = MIN;
}
dp[0][0] = 0;
for(i = 1; i <= n; i++)
{
tmax = dp[i][0]; //为tmax设置一个临时最小值
for(k = 1; k <= m && i - k >= 0; k++) //k需要遍历1~m
{
tmax = mymax(tmax, dp[i-k][0], dp[i-k][k]); //dp[i][0] 有三个动态转移方程,由第i-k分疲劳值为0(没动)和第i-k分疲劳值为k休息, 还有dp[i][0]原值
}
dp[i][0] = tmax; //找最大的即为答案
for(j = 1; j <= m; j++)
{
dp[i][j] = dp[i-1][j-1] + length[i]; //dp[i][j]转移方程只有这一个(上一分钟增加疲劳值得来)
}
}
printf("%d\n", dp[n][0]);
}
return 0;
}