【图论】Cyclic Operations—CF1867D

Cyclic Operations—CF1867D

参考资料：
大佬的提交记录
博客园的一篇文章

思路

易知 $k=1$ 时，能得到 $b$ 数组的充要条件是对于任意 $i\in [1,n]$，需要满足 $b[i]=i$。

$k>=2$ 时，我们根据 $b$ 数组中不同元素的下标和数值的关系建图（其中包含 $n$ 个点、$n$ 条边），把元素分成若干块，这些块可能是独立的点或者环。

如果一个点位于点上，那么这个点所在的块所包含的 $b$ 数组中的元素（其实就是这个元素）一定可以完成赋值。

如果一个点位于环上，那么这个环包含的点的数量只能是 $k$，这里不再具体解释。

$Code$

#include <bits/stdc++.h>
#define int long long
#define sz(a) ((int)a.size())
#define all(a) a.begin(), a.end()
using namespace std;
using PII = pair<int, int>;
using i128 = __int128;
const int N = 2e5 + 10;

int n, k;
int b[N];
int dep[N];
int ord[N];
int st[N];

void solve() {
	cin >> n >> k;
	for (int i = 1; i <= n; i ++) {
		cin >> b[i];
	}
	
	// k == 1时，对于任意i∈[1, n], 需要满足b[i] = i
	if (k == 1) {
		int ok = 1;
		for (int i = 1; i <= n; i ++) {
			ok &= (b[i] == i);
		}
		cout << (ok ? "YES\n" : "NO\n");
		return;
	}
	
	// 初始化
	int found = 1;
	for (int i = 1; i <= n; i ++) {
		dep[i] = ord[i] = st[i] = 0;
	}
	
	// [&]：按照引用表示所有封闭范围中的变量
	// 因为这个lambda表达式中用到了递归，所以需要在参数列表中加上“auto dfs”
	auto dfs = [&] (auto dfs, int u, int last, int order) {
		if (!found) {
			return;
		}
		dep[u] = dep[last] + 1; // 这个点的深度++
		ord[u] = order; // 标记这个点所在的集合编号
		st[u] = 1; // 标记这个点来过
		int v = b[u]; // b[u] = v
		if (st[v] && ord[u] == ord[v]) { // 两个点都出现且在同一个集合(出现了环)
			if (dep[u] - dep[v] + 1 != k) { // 环上点的数量 != k
				found = 0;
			}
		} else if (st[v] == 0) {
			dfs(dfs, v, u, order);
		}
	};
	
	for (int i = 1; i <= n; i ++) {
		if (st[i] == 0) {
			dfs(dfs, i, 0, i);
		}
	}
	
	cout << (found ? "YES\n" : "NO\n");
}

signed main() {
	ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
	int T = 1;
	cin >> T; cin.get();
	while (T --) solve();
	return 0;
}