CF814B An express train to reveries

注意本题输出的是排列而非序列！

考虑分类讨论，设 $c$ 表示 $\sum_{i=1}^n [a_i \neq b_i]$，即 $a_i \neq b_i$ 的 $i$ 的个数，$ans_i$ 表示答案。

若 $c = 1$，那么对于 $a_i = b_i$，则 $ans_i = a_i$，对于唯一一个 $a_i \neq b_i$，只需要从 $1$ 到 $n$ 循环，找到一个 $cnt_j$ 为 $0$ 的 $j$，那么 $ans_i = j$，$cnt_i$ 表示 $i$ 在答案排列中出现次数。

若 $c = 2$，那么对于 $a_i = b_i$，则 $ans_i = a_i$，对于两个 $a_i \neq b_i$，设这两个 $i$ 分别为 $x$ 和 $y$，分别测试 $ans_x = a_x$ 或 $ans_x = b_x$，找出可行解即可。

代码：

#pragma GCC optimize("-Ofast")
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;

const int N = 1005;

int n, a[N], b[N], cnt, cnts[N], ans[N];

inline int read()
{
	register char ch(getchar());
	register int x(0);
	while (ch < '0' || ch > '9') ch = getchar();
	while (ch >= '0' and ch <= '9')
	{
		x = (x << 1) + (x << 3) + (ch ^ 48);
		ch = getchar();
	}
	return x;
}

int main()
{
	n = read();
	for (int i = 1; i <= n; i++) a[i] = read();
	for (int i = 1; i <= n; i++)
	{
		b[i] = read();
		cnt += a[i] != b[i];
	}
	if (cnt == 1)
	{
		for (int i = 1; i <= n; i++)
		{
			if (a[i] == b[i]) ans[i] = a[i], cnts[a[i]]++;
		}
		for (int i = 1; i <= n; i++)
		{
			if (a[i] != b[i])
			{
				for (int j = 1; j <= n; j++)
				{
					if (!cnts[j])
					{
						ans[i] = j;
						break;
					}
				}
			}
		}
	}	
	else if (cnt == 2)
	{
		bool f = false;
		for (register int i = 1; i <= n; i++)
		{
			if (a[i] == b[i]) ans[i] = a[i], cnts[a[i]]++;
		}
		for (register int i = 1; i <= n; i++)
		{
			if (a[i] != b[i])
			{
				if (!cnts[a[i]])
				{
					ans[i] = a[i];
					cnts[a[i]]++;
				}
				else
				{
					if (!cnts[b[i]])
					{
						ans[i] = b[i];
						cnts[b[i]]++;
					}
					else
					{
						for (int j = i - 1; j >= 1; j--)
						{
							if (a[j] != b[j])
							{
								if (ans[j] == a[j])
								{
									cnts[a[j]]--;
									ans[j] = b[j];
								}
								else ans[j] = a[j], cnts[b[j]]--;
								break;
							}
						}
						if (!cnts[b[i]])
						{
							ans[i] = b[i];
							cnts[b[i]]++;
						}
						else if (!cnts[a[i]])
						{
							ans[i] = a[i];
							cnts[a[i]]++;
						}
					}
				}
			}
		}
	}
	for (int i = 1; i <= n; i++) printf("%d ", ans[i]);
	putchar('\n');
	return 0;
}