Game of the Rows codeforces 839B(贪心)

题目链接

题意：有n排座位，每排有2个2连座，1个4连座。有k队人。问能否让他们坐下而没有不同队伍的人相邻。
题解：先把队伍人数大于4的全部转化成小于4的，先用4连座，没有了再用2个二连座。这样就只有1,2,3的了。
先处理3的，优先用4连座，没有就用2个2连座。
在处理2的，可以用4连座，这样还可以坐一个人，这里要记录一下。还可以用2连座，还可以用两个剩下的4连座（即两个已经有两个人坐的4连座）。
1的也差不多。2连座，完整4连座，坐了人的4连座。有座就行。
考虑的情况比较多。要多想一想。

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int a[105];
int cmp(int a,int b)
{
    return a > b;
}
int main()
{
    int n,k;
    while(~scanf("%d %d",&n,&k))
    {
        memset(a,0,sizeof a);
        int sum1 = 2 * n,sum2 = n,flag = 0,sum3 = 0;
        for (int i = 1; i <= k; i++)
            scanf("%d",&a[i]);
        for (int i = 1; i <= k; i++)
        {
            while (a[i] >= 4)
            {
                if(sum2 > 0)
                {
                    sum2 --;
                    a[i] -= 4;
                }
                else if(sum1 >= 2)
                {
                    sum1 -= 2;
                    a[i] -= 4;
                }
            }
        }
        sort(a+1,a+k+1,cmp);
        for (int i = 1; i <= k; i++)
        {
            if(a[i] == 3)
            {
                if(sum2 > 0)
                {
                    sum2--;a[i] = 0;
                }
                else if(sum1 >= 2){
                    sum1 -= 2;a[i] = 0;
                }
            }
            else if(a[i] == 2)
            {
                if(sum2 > 0)
                {
                    sum2--;
                    sum3++;a[i] = 0;
                }
                else if(sum1 > 0){
                    sum1--;a[i] = 0;
                }
                else if(sum3 >= 2)
                {
                    sum3 -= 2;a[i] = 0;
                }
            }
            else if(a[i] == 1)
            {
                if(sum3 > 0)
                {
                    sum3--;a[i] = 0;
                }
                else if(sum1 > 0)
                {
                    sum1--;a[i] = 0;
                }
                else if(sum2 > 0)
                {
                    sum3++;
                    sum2 --;a[i] = 0;
                }
            }
        }
        for (int i = 1; i <= k; i++)
        {
            if(a[i] > 0)
            {
                flag = 1;
                break;
            }
        }
        if(flag)printf("NO\n");
        else printf("YES\n");
    }
}