Codeforces 1138B
——如何优雅地暴力
题目大意
马戏团有 n n n(偶数)个艺术家,我们可知道,每个艺术家是否可以作为小丑表演,是否可以作为杂技演员表演。现需将其分成2个表演节目,每个表演节目的人数相等都为 n / 2 n/2 n/2,且要求第一个表演节目中能演小丑的人数与第二个表演节目中能演杂技演员的人数相等。输出一种方案。
思路
即艺术家们有4种:00,01,10,11。
根据他们之间要求的数量关系,以及一开始的数量,可以在 O ( n 2 ) O(n^2) O(n2)范围内暴力得出。
先暴力表演第一个节目能演小丑的人数,范围是[0,min(n/2,cnt[2]+cnt[3])]
;
再假定知道表演第一个节目能演小丑的人数情况下,再暴力其中是11的艺术家人数,即11对其的贡献;
根据人数关系,可求出每种艺术家人数,不符合条件的continue
掉,找到一种方案就break
掉。
AC代码
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
const int maxn=5000;
int n;
int flag[maxn];
int cnt[4],alloc[4];
int main()
{
int x;
while(~scanf("%d",&n)){
memset(cnt,0,sizeof(cnt));
memset(alloc,0,sizeof(alloc));
for(int i=0;i<n;++i){
scanf("%1d",&x);
flag[i]=x<<1;
}
for(int i=0;i<n;++i){
scanf("%1d",&x);
flag[i]+=x;
++cnt[flag[i]];
}
bool suc=false;
for(int i=0;i<=n/2;++i){
for(int j=0;j<=i&&j<=cnt[2];++j){
if(cnt[3]<(i-j)||i<(cnt[3]-(i-j))||cnt[1]<i-(cnt[3]-(i-j))) continue;
alloc[2]=j;
alloc[3]=i-j;
alloc[1]=cnt[1]-(i-(cnt[3]-(i-j)));
if(alloc[1]>cnt[1]||alloc[1]<0) continue;
alloc[0]=n/2-alloc[2]-alloc[3]-alloc[1];
if(alloc[0]>cnt[0]||alloc[0]<0) continue;
suc=true;
goto result;
}
}
result:
if(!suc){
printf("-1\n");
continue;
}
int k[4];
memset(k,0,sizeof(k));
for(int i=0;i<n;++i){
for(int j=0;j<4;++j){
if(flag[i]==j&&k[j]<alloc[j]){
++k[j];
printf("%d%c",i+1,k[0]+k[1]+k[2]+k[3]<n/2?' ':'\n');
break;
}
}
}
}
return 0;
}
还有分类讨论贪心做法
先将10与01类型分给两个节目
- 恰好相等地分好,且11为偶数,能恰好分好,得出方案
- 恰好相等地分好但11类型为奇数,将多的那边的类型拿出1个到少的这边
- 一种多一种少,将11放在少的那边
- 恰好分好,得出方案
- 仍然不够,然后将多的那边,拿出到少的这边,即减少多的那边1贡献,增加少的这边0贡献
- 多,两边都放
- 偶数,恰好分好则得出方案
- 奇数,将多的那边的类型拿出1个到少的这边
判断人数限制n/2
没有超过,即可得出方案,反之无解