codeforces Div2 B - Block Adventure(思维题)

codeforces Div2 B - Block Adventure(思维题)

题目

http://codeforces.com/contest/1200/problem/B

题意

小明要从第1个柱子走到第n个柱子。给你以下数据：
n,m,k。n表示有n个柱子，m表示初始位置给你一个装有m个砖头的书包，k表示当两个柱子的砖头之差小于等于k时小明才能度过。
小明可以有如下操作：
如果柱子上至少有一个木块，从第i个柱子的顶部取下一个木块，放入袋子中；

如果袋子中至少有一个块，从袋子中取出一个块，放在第i列的顶部；

如果i<n并且|h_i−h_i+1|≤k，将砖头移动到i+1列的顶部。k是在游戏开始时给出的非负整数。请注意，只能移到下一列。
模拟以下题意就能过了，没啥技巧。

代码

#include<stdio.h>
#include<algorithm>
#include<math.h>
using namespace std;
int main()
{
	int T;
	scanf("%d",&T);
	while(T--)
	{
		int n,m,k;
		int h[105];
		bool flag = true;
		scanf("%d %d %d",&n,&m,&k);
		for(int i=0;i<n;i++)
			scanf("%d",&h[i]);
		for(int i=1;i<n;i++)
		{
			int p = max(0, h[i] - k);
			if(h[i-1] > p) 
			{
				m += h[i-1] - p;
			}
			else if(p - h[i-1] > m) 
			{
				flag = false;
				break;
			}
			else 
			{
				m -= (p - h[i-1]);
			}
		}
		if(flag)
			puts("YES");
		else
			puts("NO");
	}
	return 0;
}