Codeforces Round #545 (Div. 2) B. Circus 枚举 公式

题解

题目大意，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 
#include 
#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;
}