题目链接
当某个数在数组a,b内出现的次数>n时,一定无解,反之,一定能找到一个满足条件的数组b
将数组b按降序排列,若有有重复的区间,则这个区间的数一定相同。
定义一个指针it初始位置指向b[0],对于重复位置的b[i],swap(b[i], b[it])
需保证交换后a[it] != b[it] && a[i] != b[i]。由于重复区间的数相同,因此it无需回溯,即it最多只会遍历b数组一次
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 200010;
int a[N], b[N], n;
int cnt[N];
int main() {
cin >> n;
for (int i = 0; i < n; i ++) {
cin >> a[i];
cnt[a[i]] ++;
}
for (int i = 0; i < n; i ++) {
cin >> b[i];
cnt[b[i]] ++;
}
for (int i = 1; i <= n; i ++)
if (cnt[i] > n) {
cout << "No";
return 0;
}
sort(b, b + n, greater<int>());
int it = 0;
for (int i = 0; i < n; i ++) {
if (a[i] == b[i]) {
while (it < n && (b[it] == a[i] || a[it] == b[i]))
it ++;
swap(b[i], b[it]);
}
}
cout << "Yes" << endl;
for (int i = 0; i < n; i ++)cout <<b[i] <<' ';
return 0;
}