ZOJ-4033 逆向思维

题目链接：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;
}