一、题目
1、题目描述
2、输入输出
2.1输入
2.2输出
3、原题链接
1138B - Circus
二、解题报告
1、思路分析
设第一组会小丑和杂技的人数分别为x1, y1
第二组会小丑和杂技的人数分别为x2, y2
显然根据要求有:
x1 = y2 =>
x1 + x2 = x2 + y2
上式说明第二组每个人会的技能个数和 为 会小丑的总人数
贡献来自:第二组两种都会的人数,只会一种的人数
那么我们枚举两种都会的人数,就可以得到只会一种的人数,从而得到两种都不会的人数
然后检查三个人数是否合法即可
2、复杂度
时间复杂度: O(N)空间复杂度:O(N)
3、代码详解
#include <bits/stdc++.h>
using i64 = long long;
using i128 = __int128;
using PII = std::pair<int, int>;
const int inf = 1e9 + 7, P = 1e9 + 7;
/*x1/y1 x2/y2x1 = y2x1 + x2 = y2 + x2
*/
void solve() {int n;std::cin >> n;std::string s1, s2;std::cin >> s1 >> s2;int tot = 0;std::vector<std::vector<int>> cnt(3);for (int i = 0; i < n; ++ i )cnt[(s1[i] ^ 48) + (s2[i] ^ 48)].push_back(i + 1), tot += (s1[i] ^ 48);for (int i = 0; i <= cnt[2].size(); ++ i) {int j = tot - i * 2;if (j >= 0 && j <= cnt[1].size() && n / 2 - i - j >= 0 && n / 2 - i - j <= cnt[0].size()) {for (int k = i; k < cnt[2].size(); ++ k) std::cout << cnt[2][k] << ' ';for (int k = j; k < cnt[1].size(); ++ k) std::cout << cnt[1][k] << ' ';for (int k = n / 2 - i - j; k < cnt[0].size(); ++ k) std::cout << cnt[0][k] << ' ';return;}}std::cout << -1;
}int main(int argc, char** argv) {std::ios::sync_with_stdio(false), std::cin.tie(0), std::cout.tie(0);int _ = 1;// std::cin >> _;while (_ --)solve();return 0;
}