浙江省赛 ZOJ

题意:

第一行给出T代表有几个测试样例，

第二行给出n代表有几个人，

第三行给出一个由0和1组成的字符串，0代表女生，1代表男生。

并且第i个人有i个宝石。

现在要把这些人分为四组，G1 G2 两组是女生组，G3 G4两组是男生组。

分配后要满足  （G1+G3)的宝石 ==(G2+G4)的宝石

然后输出这些人被分配到哪一组的字符串。

如果有多个结果随便哪一个都行，如果没有结果输出-1。

思路：

首先把所有宝石的总和判断一下，看看是偶数还是奇数，如果是奇数肯定分不均，直接输出-1.

然后把宝石总和分一半，先给G1和G3组分配人。

也就是从第n个人开始分配组，如果G1和G3分够了宝石的一半，那剩下的就都是G2和G4的了。

#include<iostream>
#include<algorithm>
#include<sstream>
#include<string>
using namespace std;
typedef long long ll;

int T;
int main()
{
	cin >> T;
	while (T--)
	{
		ll n;
		ll half;
		ll sum;
		string num;
		string ans;

		cin >> n;
		ans.resize(n, ' ');//这个很重要，重新分配了答案字符串的大小，方便分配组
		cin >> num;

		sum = n * (n + 1) / 2;//宝石的总和
		if (sum & 1) { cout << -1 << endl; continue; }//如果是奇数就直接输出-1

		half = sum / 2;//宝石分一半
		for (int i = n; i >= 1; i--){//从后面开始分
			if (half <= n &&  ans[half - 1] == ' '){//half<=n意味着可以直接从中挑人分组了
				if (num[half - 1] == '1') {
					ans[half - 1] = '3';
				}
				else {
					ans[half - 1] = '1';
				}
				break;
			}
			half -= i;//表示已经分配掉这个人了
			if (num[i - 1] == '1') {
				ans[i - 1] = '3';
			}
			else{
				ans[i - 1] = '1';
			}
		}

		for (int i = n; i >=1; i--){
			if (ans[i - 1] == ' '){
				if (num[i - 1] == '1') {
					ans[i - 1] = '4';
				}
				else {
					ans[i - 1] = '2';
				}
			}
		}
		cout << ans << endl;
	}
	return 0;
}