codeforces #428 Div.2 B. Game of the Rows 贪心、思维

不是水题…写起来巨烦，写的时候思路断了就很不好连起来。

原题链接：
codeforces 839B

大意：
一群人坐飞机。给出飞机的排数 n ，一排 8 座和 k 个 group 。 要求不同 group 的人不能座位相邻。
如图。{1，2}，{3，4}，{4，5}，{5，6}，{7，8} 是相邻的。

给出几个有代表性的样例：

input
2 2
5 8
output
YES

input
1 2
7 1
output
NO

input
1 2
4 4
output
YES

input
1 4
2 2 1 2
output
YES

样例一

思路：
麻烦的是中间 {5，6} 相邻的情况，否则处理一下奇数即可。
首先容易想到贪心地去坐。
大于 4 人的，把 4 人的坐了，做到最大化利用。然后考虑特殊的情况 比如 aa bb0c dd
这种重叠情况通过代码实现比较烦，这里学习到一种很清晰的解法。

把 group 用优先队列处理就很方便

① 先把 4 人座坐掉。
② 再坐 2 人座，此时考虑的时候 4 人座当 2 人座，意义相同。
③ 那么 剩下来的人都是孤单的，为 1。 这些人能坐 剩下来的所有座位。可以发现剩下来的所有座位都是“2 人座（4人中的两人或是2人座）”，因为上面是按照偶数来减掉的。这样就是最优做法。

#include <bits/stdc++.h>using namespace std;typedef long long ll;#define mem(s,t) memset(s,t,sizeof(s))#define D(v) cout<<#v<<" "<<v<<endl#define inf 0x3f3f3f3f#define pb push_back//#define LOCALinline void read(int &x){    x=0;char p=getchar();    while(!(p<='9'&&p>='0'))p=getchar();    while(p<='9'&&p>='0')x*=10,x+=p-48,p=getchar();}int main() {    int n,k;    read(n);read(k);    priority_queue<int> q;    while(!q.empty()) q.pop();    for(int i=0;i<k;i++){        int x;        read(x);        q.push(x);    }    int x4=0;    while(x4<n && !q.empty()){        if(q.top()<4) break;        int t=q.top();        q.pop();        if(t/4+x4<=n){            x4+=t/4;            t%=4;            if(t) q.push(t);        }else {            t-=4*(n-x4);            x4=n;            q.push(t);        }    }    int n2=2*n+n-x4;    int x2=0;    while(x2<n2 && !q.empty()){        if(q.top()<2) break;        int t=q.top();        q.pop();        if(t/2+x2<n2){            x2+=t/2;            t%=2;            if(t) q.push(t);        }else {            t-=2*(n2-x2);            x2=n2;            q.push(t);        }    }    int n3=n-x4+n2-x2;    int x=0;    while(!q.empty()){        x+=q.top();        q.pop();    }    if(x<=n3) puts("YES");    else puts("NO");    return 0;}