ZOJ 4033 CONTINUE...?（构造）

Description

$n$个孩子，第$i$个孩子编号为$i$，给出每个孩子的性别，要求把这些孩子分成四个组，其中前两个组是女孩，后两个组是男孩，且第一个组和第三个组的孩子编号和等于第二个组和第四个组的孩子编号和，问是否有合法方案

Input

第一行一整数$T$表示用例组数，每组用例首先输入一整数$n$表示孩子数量，之后输入一个长度为$n$的$01$串表示孩子们的性别，$1$表示男孩，$0$表示女孩$(1\le n\le 10^5)$

Output

如果无解则输出$-1$，否则输出一个长度为$n$的字符串，第$i$个字符表示第$i$个孩子所属的组的编号

Sample Input

5
1
1
2
10
3
101
4
0000
7
1101001

Sample Output

-1
-1
314
1221
3413214

Solution

问题转化为把$1$~$n$分成两个和相同的集合，若$\frac{n(n+1)}{2}$为奇数显然无解，否则$n$必然可以表示成$4k$或$4k+3$的形式

如果$n=4k$，那么把前$k$个和后$k$个分为一个集合，中间$2k$个分成一个集合即可

如果$n=4k+3$，那么把$1,2$分到第一个集合，$3$分到第二个集合，后面$4k$个元素按上面的方法分即可

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=100005;
int T,n,ans[maxn];
char s[maxn];
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%s",&n,s+1);
        if(n%4!=0&&n%4!=3)printf("-1\n");
        else 
        {
            int k=n/4;
            if(n%4==0)
            {
                for(int i=1;i<=k;i++)
                    if(s[i]=='0')ans[i]=1;
                    else ans[i]=3;
                for(int i=k+1;i<=3*k;i++)
                    if(s[i]=='0')ans[i]=2;
                    else ans[i]=4;
                for(int i=3*k+1;i<=4*k;i++)
                    if(s[i]=='0')ans[i]=1;
                    else ans[i]=3;
            }
            else
            {
                for(int i=1;i<=2;i++)
                    if(s[i]=='0')ans[i]=1;
                    else ans[i]=3;
                for(int i=3;i<=3;i++)
                    if(s[i]=='0')ans[i]=2;
                    else ans[i]=4;
                for(int i=4;i<=k+3;i++)
                    if(s[i]=='0')ans[i]=1;
                    else ans[i]=3;
                for(int i=k+4;i<=3*k+3;i++)
                    if(s[i]=='0')ans[i]=2;
                    else ans[i]=4;
                for(int i=3*k+4;i<=4*k+4;i++)
                    if(s[i]=='0')ans[i]=1;
                    else ans[i]=3;
            }
            for(int i=1;i<=n;i++)printf("%d",ans[i]);
            printf("\n");
        }
    }
    return 0;
}