Justice
题意:给定n个k,k代表1/2^k,问最后能否选择一些数的和使得有两个数大于等于1/2
思路:要构成一个1/2, 需要1个1,2个2,4个3...也可以是1个2+2个3...那么把k从小到大排序,设定一个p代表现在需要的数,cnt1,cnt2,表示两份组成1/2需要p的数量。我们只管最后能组成两份1/2,其他的不管。学个数据模拟一下。
3 3 3 3 4 4 4 4 5 6
首先p=1,cnt1=1,cnt2=1;
第一个碰到3,需要4个。
p=3,cnt1=4,cnt2=4.
此时有一个满足情况了,
然后碰到4,
p=4,cnt1=0,cnt2=8.
把4个4用完后,p=4,cnt1=0,cnt2=4;
再碰到5,
p=5,cnt1=0,cnt2=8,因为现在只需要8个5就可以组成1/2了,前面已经有了1/4.
再碰到6
p=6,cnt1=0,cnt2=14,1/4+1/32=18/64,只需要14个6就可以了。
所以我们通过最后看cnt1,cnt2是否等于0来判断能否组成两个2,模拟一下即可。
#include<bits/stdc++.h>
using namespace std;
struct node{
int v,id;
}a[100005];
int cmp(node x,node y){
return x.v<y.v;
}
int ans[100005];
int main(){
int T;
scanf("%d",&T);
for(int i=1;i<=T;i++){
int n;
scanf("%d",&n);
int cnt1=1,cnt2=1,pre=1;
memset(ans,0,sizeof(ans));
for(int j=1;j<=n;j++){
scanf("%d",&a[j].v);
a[j].id=j;
}
sort(a+1,a+1+n,cmp);
int f=0;
for(int j=1;j<=n;j++){
while(cnt1+cnt2<=n-j+1&&pre<a[j].v){
cnt1*=2;
cnt2*=2;
pre++;
}
if(cnt1+cnt2>n-j+1){
f=1;
break;
}
if(cnt1){
cnt1--;
ans[a[j].id]=1;
}
else {
cnt2--;
}
if(!cnt1&&!cnt2)
break;
}
if(f||cnt1||cnt2){
printf("Case %d: NO\n",i);
}
else{
printf("Case %d: YES\n",i);
for(int j=1;j<n;j++)
printf("%d",ans[j]);
printf("%d\n",ans[n]);
}
}
}