1175. 【动态规划】黑熊过河

题目描述

晶晶的爸爸给晶晶一道难题：有一只黑熊想过河，但河很宽，黑熊不会游泳，只能借助河面上的石墩跳过去，他可以一次跳一墩，也可以一次跳两墩，但是每跳一次都会耗费一定的能量，黑熊最终可能因能量不够而掉入水中。

所幸的是，有些石墩上放了一些食物，这些食物可以给黑熊增加一定的能量。问黑熊能否利用这些石墩安全的抵达对岸，请计算出抵达对岸后剩余能量的最大值。

输入

第一行包含两个整数 P (黑熊的初始能量)，Q (黑熊每次起跳时耗费的能量)，0≤P,Q≤1000；

第二行只有一个整数 n (1≤n≤10^6），即河中石墩的数目；

第三行有 n 个整数，即每个石墩上食物的能量值 ai (0≤ai≤1000)。

输出

仅一行，若黑熊能抵达对岸，输出抵达对岸后剩余能量的最大值，若不能， 则输出 NO。

样例数据

输入 #1 复制

12 5
5
0 5 2 0 7

输出 #1 复制

6

提示

注意黑熊一开始不在石墩上，且最后也要从石墩上跳上河对岸。

一道经典DP题。

思路：

用f[i]表示跳到第i块石墩上的最大能量值，初始状态：f[0]=P,f[1]=P-Q+第一块石墩上的能量，

状态转移方程：f[i]=f[i-1]+f[i-2]。

代码：

#include <bits/stdc++.h>
using namespace std;
int a[1000005],f[1000005];
int main()
{
	int x,y,n;
	cin>>x>>y;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
	}
	f[0]=x;
	f[1]=x-y+a[1];
	for(int i=2;i<=n+1;i++)
	{
		f[i]=(max(f[i-1],f[i-2])-y)+a[i];
	}
	int t=1;
	for(int i=1;i<=n;i++)
	{
		if(f[i]<y)
		{
			if(t==0)
			{
				cout<<"NO";
				return 0;
			}
			else t=0;
		}
		else t=1;
	}
	cout<<f[n+1];
	return 0;
}