大致题意
给 n 个砝码 ,每个重量为 2 ^ (-ai) ,问能否把这 n 个砝码分成两堆,使得两堆重量都大于等于 1/2 。
思路
首先说一个标程思路,用一个最大堆,每次取出前两个,如果两个相等,push(x-1),否则把大的那个扔掉,因为也没办法和出来了,最后看是否剩下多于两个1。
训练赛的时候没想到这么做,当时的做法是,把 2^0 的都转成 2 ^ (-1),然后用一个 2 ^ (-1) 去数组里从前往后减,往后一次就要把剩余要消减的值乘2,直到把这个 2 ^(-1) 减成 0 。由于一直向后乘2 的过程中,剩余值可能很大,所以当 now > n的时候就不用往下做了,因为总共 n 个值,就算是 2 ^ (-n) 也要 n个才行,所以后面一定不可能凑出 2 ^ (-1)了。做完一遍之后,如果能凑出 2 ^(-1) ,那么再做一遍这个操作,如果还能凑出一个 2 ^(-1) ,那么就可行。复杂度 O(n)。
代码
#include<bits/stdc++.h>
using namespace std;
#define maxn 100005
#define maxm 1000006
#define ll long long int
#define INF 0x3f3f3f3f
#define inc(i,l,r) for(int i=l;i<=r;i++)
#define dec(i,r,l) for(int i=r;i>=l;i--)
#define mem(a) memset(a,0,sizeof(a))
#define sqr(x) (x*x)
#define inf (ll)2e18+1
#define mod 998244353
int read(){
int x=0,f=1;char ch=getchar();
while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
return f*x;
}
int T,n,a[maxn],b[maxn],vis[maxn];
void init()
{
inc(i,1,n){
if(b[i]<maxn)vis[b[i]]=a[b[i]]=0;
}
}
int main()
{
T=read();
int kase=0;
while(T--){
int ma=0;
n=read();
inc(i,1,n){
b[i]=read();
if(b[i]>=maxn)continue;
ma=max(b[i],ma);
a[b[i]]++;
}
printf("Case %d: ",++kase);
a[1]+=2*a[0];
bool flag=0;
int now=1;
inc(i,1,ma){
int tmp=min(now,a[i]);
if(now>n)break;
now-=tmp;
a[i]-=tmp;
vis[i]=tmp;
if(now==0){
flag=1;
break;
}
now*=2;
}
if(flag==0){printf("NO\n");init();continue;}
flag=0;
now=1;
inc(i,1,ma){
if(now>n)break;
int tmp=min(now,a[i]);
now-=tmp;
a[i]-=tmp;
if(now==0){
flag=1;
break;
}
now*=2;
}
if(flag==0){printf("NO\n");init();continue;}
printf("YES\n");
inc(i,1,n){
if(b[i]<maxn&&vis[b[i]]>0){printf("1");vis[b[i]]--;}
else printf("0");
}
printf("\n");
init();
}
return 0;
}