CF-Round #629-div3-C题

CF-Round #629-div3-C题

C. Ternary XOR

传送门

这道题是道贪心题

题目背景给你三进制的异或操作：(a[i] + b[i]) % 3 = x[i]；
给你x序列，问a, b序列，要求max(a, b)最小

因为x的一个数字一定是2（题目中说的）
并且不能有前导0
所以a[0] = b[0] = 1；毫无疑问

我们贪心策略如下：
当x[i] = 2；我们可以把2分配为1， 1这样才能保证最大值最小不是吗
x[i] = 0；都为0即可
那我们碰到1怎么办呢。
当我们碰到第一个1的时候让第一个序列为1，第二个序列为0
至此以后所有的数字，第一个序列之后全是0，第二个序列之后全部等于x[i]；
这样可以使得最大值最小
想想：第二个序列某个地方从0开始的，肯定小，第一个序列某个地方从1开始的，显然这个时候第一个序列已经比第二个序列大了，所以我们现在需要第二个序列尽可能小，所以都赋值为0（记住！！这个时候不管怎么样第二个序列是两者之间最大的了）

代码部分~

#include <bits/stdc++.h>
using namespace std;
 
string s;
int n;
 
int main()
{
	int t;
	cin >> t;
	while (t--)
	{
		cin >> n;
		cin >> s;
		string a(n, '0');
		string b(n, '0');
		a[0] = '1';
		b[0] = '1';
		int flag = 0;
		for (int i = 1; i < n; i++)
		{
			if (flag)
			{
				a[i] = '0';
				b[i] = s[i];
			}
			else if (s[i] == '1' && !flag)
			{
				a[i] = '1';
				b[i] = '0';
				flag = 1;
			}
			else if (!flag && s[i] == '0')
			{
				a[i] = b[i] = '0';
			}
			else if (!flag && s[i] == '2')
			{
				a[i] = b[i] = '1';
			}
		}
		cout << a << endl << b << endl;
	}
	return 0;
}