前情提要:今天搞了个重现赛,一个人打,AB先水过了以后,卡在了C题好久,然后放弃了。想起来了又看了一遍,在博客上看题解看了好久才看懂。
题意:给n个数字,要求分成两组,另两组内的数字按照他所定义的运算之和至少是
1
2
\frac{1}{2}
21,首先能想到的是,只要找出来两组数字,让各自组内相加之后得到
1
2
\frac{1}{2}
21即可。可以发现,一个1可以分成两个2,一个2可以分成两个3,所以就有了以下的方法:
首先把所有的数字从小到大排序(肯定是优先选取较大的数字),接着定义一个当前需要的数字k,初始化为1,表示当前两组分别需要一个1,然后定义两个计数器cnt1,cnt2表示两组还分别需要多少个当前的数字。由于数字已经排好序,所以直接按顺序枚举数字,如果第 i 个数字比当前的数字(k)大,那么表示不会再出现数字k了。所以可以直接将当前数字+1,并且cnt1和cnt2翻倍,表示需要两倍数目的当前数字才能达到目标。剩下的就是一些特判了,在代码注释中给出。
#include<cstdio>
#include<algorithm>
#include<iostream>
using namespace std;
struct d {
int k, id;
}a[100010];
int cmp(d a, d b)
{
return a.k < b.k;
}
int main()
{
int T;
scanf("%d", &T);
for (int q = 1; q <= T; q++)
{
int n;
scanf("%d", &n);
for (int i = 1; i <= n; i++)
{
scanf("%d", &a[i].k);
a[i].id = i;
}
sort(a + 1, a + n + 1, cmp);
int k = 1;
int cnt1 = 1, cnt2 = 1;
int flag = 1;
int vis[100010];
memset(vis, 0, sizeof(vis));
for (int i = 1; i <= n; i++)
{
while (cnt1 + cnt2 <= n - i + 1 && k < a[i].k)//前提条件是,剩下的数字还够,即两组所需要
{ //的数字的总和小于等于剩下的数字
cnt1 <<= 1;
cnt2 <<= 1;
k++;
}
if (cnt1 + cnt2 > n - i + 1)//数字不够了直接退出,输出NO
{
flag = 0;
break;
}
if (cnt1)//如果while循环中结束了,要么表示数字不够,要么是出现了符合当前所需数字的情况出现,
{ //所以直接将当前数字划分给第一组,第一组如果已经足够了,那么划分给第二组。
cnt1--;
vis[a[i].id]=1;
}
else cnt2--;
if (cnt1 == 0 && cnt2 == 0) break;
}
printf("Case %d: ", q);
if (flag||cnt1||cnt2)//一个坑,两组数中任何一组没有被清零都表示无法完成目标。
{
printf("YES\n");
for (int i = 1; i <= n; i++)
{
printf("%d", vis[i]);
}
printf("\n");
}
else
{
printf("NO\n");
}
}
}