蓝桥杯：VIP试题 算法训练 找零钱

蓝桥杯：VIP试题 算法训练 找零钱
资源限制
时间限制：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

#include<iostream>
#include<string>
using namespace std;


int main(){
	long n; 
	int flag = 1;
	int *a = new int[1000001];
	int b[2] = {0,0}; //表示当前25元和50元的纸张数 ,b[0]:25 b[1]:50 
	cin>>n;
	for(int i = 1;i <= n;i++){
		cin>>a[i];
	}
	for(int i = 1;i <= n;i++){
		if(a[1] == 50 ||a[1] == 100){//如果第一张是50or100则无法找零 
			cout<<"NO"; 
			flag = 0;
			break; 
		} 
		/*从第i张开始 无法找零的情况
		 
		1.当前收到50元时，没有25元找
		2.当前收到100元时：
			①没有50元现钞并且25元的数目少于三张
			②有50元现钞但是没有25元现钞 
		
		*/
		if((a[i]==50 && b[0] == 0)||(a[i]==100 && ((b[0] < 3&&b[1]==0) || (b[1]>0 &&b[0]==0)))){
			cout<<"NO"; 
			flag = 0;
			break;
		}
		if(a[i] == 25){
			b[0]++;
			flag = 1;
		}else if(a[i] == 50){
			b[1]++;
			b[0]--;
			flag = 1;
		}else if(a[i] == 100){
			b[0]--;
			b[1]--;
			flag = 1;
		}
		 
	}
	if(flag == 1){
		cout<<"YES";
	}

	delete[] a;
	return 0;
}