浙江省赛 ZOJ - 4033
题意:
第一行给出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; }