试题 算法训练 找零钱 C++

试题 算法训练 找零钱
资源限制
时间限制：1.0s 内存限制：256.0MB
问题描述
　　有n个人正在饭堂排队买海北鸡饭。每份海北鸡饭要25元。奇怪的是，每个人手里只有一张钞票（每张钞票的面值为25、50、100元），而且饭堂阿姨一开始没有任何零钱。请问饭堂阿姨能否给所有人找零（假设饭堂阿姨足够聪明）
输入格式
　　第一行一个整数n，表示排队的人数。

接下来n个整数a[1],a[2],…,a[n]。a[i]表示第i位学生手里钞票的价值（i越小，在队伍里越靠前）
输出格式
　　输出YES或者NO
样例输入
4
25 25 50 50
样例输出
YES
样例输入
2
25 100
样例输出
NO
样例输入
4
25 25 50 100
样例输出
YES
数据规模和约定
　　n不超过1000000

**我的思路：**定义两个变量a,b（我定义的是数组），如果第一个进来的是25，就第一个变量a++，50的话就b++，然后判断a是否大于0(因为要找零需要-1)，如果大于0就证明找零成功，不大于0就证明找零失败，直接输出no，然后判断输入100的，因为100不用找，所以就不需要+1，首先判断a>0并且b>0,因为你可能后面找的会没零钱，先找50的，50没有再找25的，或者判断a>2，就是问你25大不大于三张，这样也可以找钱成功，就这两种情况了，都不满足那就找零失败，输出no，然后继续下一次输入，如果这些都没问题，那就证明没有找零失败，最后输出yes.
写的很啰嗦，有兴趣可以耐心看一下。
下面是代码：

#include<iostream>
using namespace std;
int main(){
	int a[2]={0};
	int n,b;
	cin>>n;
	for(int i=0;i<n;i++){
		cin>>b;
		if(b==25)
		{
			a[0]++;	
		}else if(b==50){
			a[1]++;
			if(a[0]==0){
				cout<<"NO";
				return 0;
			}
			a[0]--;
		}else{
			if(a[1]>0&&a[0]>0){
				a[1]--;
				a[0]--;
			}else if(a[0]>2){
				a[0]-=3;
			}else{
				cout<<"NO";
				return 0;
			}
			}
		}
		cout<<"YES";
		return 0;
	}