不是水题…写起来巨烦,写的时候思路断了就很不好连起来。
原题链接:
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 LOCAL
inline 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;
}