ZOJ - 4033 CONTINUE...?

题目链接 ┗|｀O′|┛ 嗷~~

有n个人，第i个人的权值为i，有男女之分，先将男生和女生分成两组，其中一组女生和一组男生权值的和与另外两组的和相等，分组允许为空，求分组方案。

---

首先求总的权值是否为偶数，如果是继续进行；不是直接输出。
是偶数的话这里需要一个辅助结论，小于∑n 的数，一定可以由{1,2,3…n}的子序列表示。所以只要是偶数就必然有答案，然后直接向下减就可以了。

证明：
对于序列{1,2,3}，包括6在内的所有数都可以由这个子序列表示。再考虑序列{1,2,3,4}，相当于序列{1,2,3}​+4,故5-10都可被表示，而4<6,所以1-4也可被表示。不断推广下去原命题得证。

---

#include <stdio.h>
#include <climits>
#include <cstring>
#include <time.h>
#include <math.h>
#include <iostream>
#include <algorithm>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <utility>
#include <vector>
#include <string>

#define INF 0x3f3f3f3f
#define ll long long
#define Pair pair<int,int>
#define re return

#define getLen(name,index) name[index].size()
#define mem(a,b) memset(a,b,sizeof(a))
#define Make(a,b) make_pair(a,b)
#define Push(num) push_back(num)
#define rep(index,star,finish) for(register int index=star;index<finish;index++)
#define drep(index,finish,star) for(register int index=finish;index>=star;index--)
using namespace std;
const int maxn=1e5+5;

int N,M;
char sex[maxn];
bool used[maxn];
int ans[maxn];
int main(){
    int _;
    scanf("%d",&_);
    while(_--){
        scanf("%d",&N);
        scanf("%s",sex+1);

        mem(used,false);
        ll sum=1LL*(1+N)*N/2;
        if(sum%2){
            printf("-1\n");
            continue;
        }

        ll target=sum/2;
        ll pos=N;
        while(target>0){
            target-=pos;
            used[pos]=true;
            if(sex[pos]-'0'){   //boy
                ans[pos]=4;
            }else{
                ans[pos]=2;
            }
            pos=min(target,pos-1);
        }
        rep(i,1,N+1){
            if(used[i])
                continue;
            if(sex[i]-'0'){
                ans[i]=3;
            }else{
                ans[i]=1;
            }
        }

        rep(i,1,N+1){
            printf("%d",ans[i]);
        }
        printf("\n");

    }
    re 0;
}