Trouble Sort（思维）

传送门

题目要求是将两种数字进行排序

其实该问题的解存在两种情况

情况一：

• 全为种类0或种类1，这样由于不能交换，只需要判断是否有序就可以了

情况二：

• 不难发现，两种都存在，那么一定可以有序，假设有n-1个种类1的，有1个种类0的，那么这一个种类0的数字可以将n-1个种类为1的数字交换成有序的，而具体交换过程我们不需要考虑。当有更多0时同理。

#include <bits/stdc++.h>
using namespace std;

const int N = 505;
int a[N], b[N];
vector<int> vec1, vec0;
void solve(){
	vec1.clear();
	vec0.clear();
	int n;
	cin >> n;
	for(int i = 1; i <= n; i ++) cin >> a[i];
	for(int i = 1; i <= n; i ++) cin >> b[i];
	bool flag0 = false, flag1 = false;
	for(int i = 1; i <= n; i ++){
		if(b[i] == 0)
			vec0.push_back(a[i]);
		else
			vec1.push_back(a[i]);
		if(b[i] == 0)
			flag0 = true;
		else
			flag1 = true;
	}
	int size0 = vec0.size();
	int size1 = vec1.size();
	if(flag0 && !flag1 || !flag0 && flag1){
		if(flag0){
			for(int i = 1; i < size0; i ++){
				if(vec0[i] < vec0[i - 1]){
					cout << "NO" << endl;
					return;
				}
			}
			cout << "YES" << endl;
			return;
		}else{
			for(int i = 1; i < size1; i ++){
				if(vec1[i] < vec1[i - 1]){
					cout << "NO" << endl;
					return;
				}
			}
			cout << "YES" << endl;
			return;
		}
	}
	if(flag0 && flag1)
		cout << "YES" << endl;
}
int main(){
	int t;
	cin >> t;
	while(t --){
		solve();
	}
	
	return 0;
}