题解
题目大意,n个人,有些可以表演小丑有些可以表演杂技,要求平分为两组,第一组可表演小丑的数量等于第二组可表演杂技的数量。
n不是很大O(N^2)复杂度可以通过,什么都不能表演的记为a,只能表演小丑记为b,只能表演杂技记为c,都可以记为d。
枚举第一队b的数量和d的数量,根据第一组可表演小丑数量等于第二组可表演杂技数量可得方程b + d == (cn - c) + (dn - d)。
cn, dn表示c和d的总数,移项得c = cn + dn - b - d * 2,则可以通过枚举的b和d得到c。
每队总人数等于n/2则可得到a = n / 2 - b - c - d,最后判断ac的合法性即可。
AC代码
#include <stdio.h>
#include <bits/stdc++.h>
#define fst first
#define sed second
using namespace std;
typedef long long ll;
const int INF = 0x3f3f3f3f;
const ll LINF = 0x3f3f3f3f3f3f3f3f;
const int N = 5e3 + 10;
char s[N], t[N];
int main()
{
#ifdef LOCAL
//freopen("C:/input.txt", "r", stdin);
#endif
int n;
cin >> n;
scanf("%s%s", s + 1, t + 1);
vector<int> va, vb, vc, vd;
for (int i = 1; i <= n; ++i)
{
s[i] -= '0', t[i] -= '0';
if (!s[i] && !t[i]) va.push_back(i);
else if (s[i] && !t[i]) vb.push_back(i);
else if (!s[i] && t[i]) vc.push_back(i);
else vd.push_back(i);
}
for (int b = 0; b <= vb.size(); ++b) //第一组b的数量
for (int d = 0; d <= vd.size(); ++d) //第一组d的数量
{
int c = vc.size() + vd.size() - b - d * 2; //第一组可表演小丑=第二组可表演杂技 b + d == (vc.size() - c) + (vd.size() - d) 移项
if (c >= 0 && c <= vc.size())
{
int a = n / 2 - b - c - d; //总人数为n/2
if (a >= 0 && a <= va.size())
{
while (a) printf("%d ", va[--a]);
while (b) printf("%d ", vb[--b]);
while (c) printf("%d ", vc[--c]);
while (d) printf("%d ", vd[--d]);
cout << endl;
exit(0);
}
}
}
cout << -1 << endl;
return 0;
}