蓝桥杯 ALGO-232 算法训练 找零钱

算法训练 找零钱

时间限制：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 <stdio.h>

int main()
{
    int n;
    int a[1000005] = { 0 };

    scanf("%d", &n);
    for (int i = 0; i < n; ++i)
        scanf("%d", &a[i]);

    int num_25 = 0, num_50 = 0;
    int failed = 0;
    for (int i = 0; i < n; ++i)
    {
        if (a[i] == 25)
            num_25++;
        else if (a[i] == 50)
        {
            num_50++;
            if (num_25 > 0)
                num_25--;
            else
            {
                failed = 1;
                break;
            }
        }
        else
        {
            if (num_25 > 0 && num_50 > 0)
            {
                num_25--;
                num_50--;
            }
            else if (num_25 >= 3)
                num_25 -= 3;
            else
            {
                failed = 1;
                break;
            }
        }
    }
    if (failed)
        printf("NO");
    else
        printf("YES");

    return 0;
}