知识点:并查集
难度:4
做了很多这种类型的题目,所以现在做这种题比较顺手了,这种类型的题目就是可以任意交换一个序列上的给定两个位置的元素,能否由现在的序列得到目标序列,可以看到限制还是比较少的,是交换任意的次数,这种题一般都是用的图论的模型,可以交换某两个位置的元素,那么就是这两个位置的下标之间有一条边,因次一个序列可以得到若干个连通块,如果两个序列相应的连通块里面的元素总的是相同的,不需要考虑具体的位置,那么这两个序列是可以通过交换得到的,
对于这道题,更一般的模型可能是,给定两个序列,不一定是全排列,也就是有重的,值任意的,然后题目给定条件,可以交换某些两个位置的元素,可以交换任意的次数,问能否完成两个序列之间的转化,那么思路就是通过下标建立连通块,然后遍历两个序列,分别往连通块里面放值,然后按照下标的顺序遍历两个放过值的连通块,看看是不是每个连通块里面的值总体都是相等的,这里应该需要用到排序,因为可能有重值,这个样子可能麻烦一点但是这是一般性的解题思路,但是这个题是比较特殊的,因为首先它的序列是一个全排列,这就很特殊了,第二个特殊的点是它的目标序列就是递增的,就是下标的排列顺序,所以这个题可以很简单的解决,用下标建立的连通块就像是一个图的结构,还没有往里面填值,上面说的普通的模型是要填两次,把得到的两个图拿来比较,只要每个连通块里面值总体相同(不知道怎么说,就这么说了,应该能明白是什么意思),就行了,这个题就相当于,建立好了图的结构,同时每个连通块里面存的数字就是下标,这就是目标序列的特殊之处,再加上这个是全排列是没有重复的,可以像我那样直接取比较,没有重复的是可以直接比较的一个原因,因为没有重复的那就是每个数只有一个,不在应该在的连通块就是错的,有重复的显然不能这样子判断
对于这个题还有一个小注意的点,那就是这个序列不是环形的,可能向前向后没有可以交换的位置
#include <bits/stdc++.h>
using namespace std;
const int N = 105;
int fa[N];
int get(int x) {
if (x == fa[x]) return x;
return fa[x] = get(fa[x]);
}
void merge(int x, int y) {
fa[get(x)] = get(y);
}
int main() {
for (int i = 0; i < N; i++) fa[i] = i;
int n;
cin >> n;
int a[N], b[N];
for (int i = 1; i <= n; i++) cin >> a[i];
for (int i = 1; i <= n; i++) {
cin >> b[i];
int x = i, y = i;
if (i - b[i] >= 1) x -= b[i];
if (i + b[i] <= n) y += b[i];
merge(i, x);
merge(i, y);
}
int ok = 1;
for (int i = 1; i <= n; i++) {
if (get(i) != get(a[i])) ok = 0;
}
cout << (ok ? "YES" : "NO");
return 0;
}
下面是应对一般情况的写法,本题作为特殊情况当然也能通过,就是因为排序,效率稍微慢一点
#include <bits/stdc++.h>
using namespace std;
const int N = 105;
int fa[N];
int get(int x) {
if (x == fa[x]) return x;
return fa[x] = get(fa[x]);
}
void merge(int x, int y) {
fa[get(x)] = get(y);
}
int main() {
for (int i = 0; i < N; i++) fa[i] = i;
int n;
cin >> n;
int a[N], b[N];
for (int i = 1; i <= n; i++) cin >> a[i];
for (int i = 1; i <= n; i++) {
cin >> b[i];
int x = i, y = i;
if (i - b[i] >= 1) x -= b[i];
if (i + b[i] <= n) y += b[i];
merge(i, x);
merge(i, y);
}
vector<int> v1[N], v2[N];
for (int i = 1; i <= n; i++) {
v1[get(i)].push_back(i);
v2[get(i)].push_back(a[i]);
}
for (int i = 1; i <= n; i++) {
sort(v1[i].begin(), v1[i].end());
sort(v2[i].begin(), v2[i].end());
}
int h[N] = {};
int ok = 1;
for (int i = 1; i <= n && ok; i++) {
int x = get(i);
if (!h[x]) {
h[x] = 1;
for (int j = 0; j < (int) v1[x].size() && ok; j++) {
if (v1[x][j] != v2[x][j]) ok = 0;
}
}
}
cout << (ok ? "YES" : "NO");
return 0;
}