题目链接:https://vjudge.net/problem/ZOJ-4033
题意:给你一个字符串,0代表女生,1代表男生,每个人的能力值是其下标+1,女生只能分在1,2组,男生分在3,4组。问能否分组让1,3组的总能力值等于2,4组。若无解输出-1。
思路:先统计总的能力值num,若为奇数则一定无解。若为偶数,则可以让男生与女生的能力值相等,即1,2组与3,4组分别凑出num/2。注意只能从后往前遍历,因为从前往后不一定能凑出来。遍历时若num/2大于当前能力值,则可以做标记。最后遍历一遍将带标记的分为1,3组。其实可以优化,在凑num/2时就可以做出分组。
坑点:注意longlong
#include <bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
int vis[maxn];
int main()
{
ios::sync_with_stdio(false);
int T;
cin>>T;
while(T--)
{
long long l;
cin>>l;
string s;
cin>>s;
long long num=(l+1)*l/2;
if(num%2==1)
{
cout<<"-1"<<endl;
continue;
}
memset(vis,0,sizeof(vis));
num/=2;
for(long long i=l-1; i>=0; i--)
{
if(num>=i+1)
{
num=num-i-1;
vis[i]=1;
}
else
{
continue;
}
if(num==0)
{
break;
}
}
for(int i=0; i<l; i++)
{
if(vis[i])
{
if(s[i]=='0')
{
s[i]='1';
}
else
{
s[i]='3';
}
}
else
{
if(s[i]=='0')
{
s[i]='2';
}
else
{
s[i]='4';
}
}
}
cout<<s<<endl;
}
return 0;
}