黑熊过河

//动态规划-基础

题目描述

晶晶的爸爸给晶晶出了一道难题：有一只黑熊想过河，但河很宽，黑熊不会游泳，只能借助河面上的石墩跳过去。它可以一次跳一墩，也可以一次跳两墩，但是每跳一次都会耗费一定的能量，黑熊最终可能因能量不够而掉入水中。所幸的是，有些石墩上放了一些食物，些食物可以给黑熊增加一定的能量。问黑熊能否利用这些石墩安全地抵达对岸？请计算出抵达对岸后剩余能量的最大值。

输入输出格式

输入格式：

第1行包含两个整数P和Q（0≤P,Q≤1000），其中P表示黑熊的初始能量，Q表示黑熊每次起跳时耗费的能量。

第2行只有一个整数n（1≤n≤10^6），表示河中石墩的数目。

第3行有n个整数，表示每个石墩上食物的能量值ai（0≤ai≤1000）。

输出格式：

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

输入输出样例

输入样例#1：

12 5
5
0 5 2 0 7

输出样例#1：

6

$AC$代码：

#include
using namespace std;
int n,q,p,sub,a[1000005],f[1000005];
int main()
{
	cin>>p>>q>>n;
	//其中P表示黑熊的初始能量，Q表示黑熊每次起跳时耗费的能量。
	for(int i=1;i<=n;i++)
		cin>>a[i];
	f[0]=p;//初始能量
	for(int i=1;i<=n+1;i++)
	{
		sub=max(f[i-1],f[i-2]);//前两个，注:每次最多跳2格 
		if(sub>=q) f[i]=sub-q+a[i];//剩余能量  
		else
		//if(sub
		{
			cout<<"NO"<<endl;     //因能量不够而掉入水中
			return 0;//结束程序 
		}
	}
	cout<<f[n+1]<<endl;    //最后能量 
	return 0;
}