Atcoder256 E

  题意：给我们一个序列，表示第$i$个人想要在$X_i$人前面获得糖果，不然就会获得不满意度。要我们求出来这个序列中的最小不满意度是多少。
  思路：因为这个题假设每一个人入度都为$1$的话，它整体就是一个环，所以现在去考虑序列中存在入度为$0$的点。我们考虑用$DSU$来存储边得关系

#include <bits/stdc++.h>

using i64 = long long;

int main() {
	std::cin.tie(nullptr)->sync_with_stdio(false);

	int n; std::cin >> n;

	std::vector<int> a(n + 1), f(n + 1), c(n + 1);
	std::iota(f.begin(), f.end(), 0);

	for (int i = 1; i <= n; i ++ ) std::cin >> a[i];
	for (int i = 1; i <= n; i ++ ) std::cin >> c[i];

	std::function<int(int)> find = [&] (int x) -> int {
		return x == f[x] ? f[x] : f[x] = find(f[x]);
	};

	auto same = [&] (int u, int v) -> bool {
		return find(u) == find(v);
	};

	auto merge = [&] (int u, int v) -> void {
		int fu = find(u), fv = find(v);
		if (fu == fv) return ;
		f[fv] = fu;
	};

	i64 ans = 0;
	for (int i = 1; i <= n; i ++ ) {
		if (!same(i, a[i])) {
			merge(i, a[i]);
			continue;
		}
		int cur = c[i], v = i;
		do {
			v = a[v];
			cur = std::min(cur, c[v]);
		}while(v != i);
		ans += cur;
	}
	std::cout << ans << "\n";
	return 0;
}

__EOF__

本文作者：HoneyGrey
本文链接：https://www.cnblogs.com/Haven-/p/16417589.html
关于博主：评论和私信会在第一时间回复。或者直接私信我。
版权声明：本博客所有文章除特别声明外，均采用 BY-NC-SA 许可协议。转载请注明出处！
声援博主：如果您觉得文章对您有帮助，可以点击文章右下角【推荐】一下。您的鼓励是博主的最大动力！