The 15th Zhejiang Provincial Collegiate Programming Contest Sponsored by TuSimple - J

J、CONTINUE…?
链接：http://acm.zju.edu.cn/onlinejudge/showContestProblem.do?problemId=5759
题意：长为n由0,1组成的字符串，Ai都有对应的权值i（1<=i<=n），0只能分为1,2组，1只能分为3，4组，要使第一组与第三组的权值和跟第二组与第四组的权值和相等，输出对应的一种分组方案。若无方案，输出-1.

思路：
两个组的权值和相等，那么总的权值和必为偶数，否则输出-1.
设总的权值和为sum,要使两个分组的权值和为sum/2，令tmp=1+2+……+k，并且前k个字符已做标记，假设tmp是比sum/2小的最大的那个数，那么此时有两种情况：
1.tmp+k+1==sum/2时，对第k+1个字符做标记
2.tmp+k+1>sum/2，对第k+1个字符做标记，将已经标记过的tmp+k+1-sum/2个字符的标记去掉

最后根据标记输出就可以了。

代码：

#include <bits/stdc++.h>
using namespace std;

#define mem(a,n) memset(a,n,sizeof(a))
#define rep(i,a,n) for(int i=a;i<n;i++)

typedef long long ll;
const double eps=1e-5;
const int N=1e5+5;
const int MOD=1e7+7;
const int INF=0x3f3f3f3f;
const int dir[4][2]= {1,0,-1,0,0,1,0,-1};

char a[N];
bool vis[N];
int main()
{
    ll T,n;
    scanf("%lld",&T);
    while(T--)
    {
        scanf("%lld %s",&n,a+1);
        for(int i=1; i<=n; i++)
        {
            vis[i]=0;
        }
        ll ss=n*(n+1)/2,sum=0;///注意计算时中间结果会溢出  即爆int
        if(ss%2==1)
        {
            puts("-1");
            continue;
        }
        ss/=2;
        for(int i=1; i<=n; i++)
        {
            if(sum+i<=ss)
            {
                sum+=i,vis[i]=1;
                if(sum==ss) break;
            }
            else
            {
                vis[i]=1;
                vis[sum+i-ss]=0;
                break;
            }
        }
        for(int i=1; i<=n; i++)
        {
            if(vis[i])
            {
                if(a[i]=='0') putchar('1');
                else putchar('3');
            }
            else
            {
                if(a[i]=='0') putchar('2');
                else putchar('4');
            }
        }
        puts("");
    }
    return 0;
}