牛客练习赛41 C 抓捕盗窃犯 并查集

题解

把小偷行走的路线看作一个有向图，如果是一个环则整个环内任意选一个就可以得到整个环的收益。
把每个环看作一个点，如果是一个链则选取链出度为0节点则可以得到整个链的收益。
由于每个点出度至多唯一，使用并查集求出每个链的收益总和，选取前m大即可，注意m可能少于链的数量。

AC代码

#include <stdio.h>
#include <bits/stdc++.h>
#define fst first
#define sed second
using namespace std;
typedef long long ll;

const int INF = 0x3f3f3f3f;
const ll LINF = 0x3f3f3f3f3f3f3f3f;
const int N = 1e5 + 10;
int vex[N];
ll sum[N];

int find(int x)
{
	return vex[x] == 0 ? x : vex[x] = find(vex[x]); //vex[x] == 0
}
void join(int x, int y)
{
	x = find(x), y = find(y);
	if (x != y)
	{
		sum[y] += sum[x];
		vex[x] = y;
	}
}
int main()
{
#ifdef LOCAL
	//freopen("C:/input.txt", "r", stdin);
#endif
	int n, m;
	cin >> n >> m;
	for (int i = 1; i <= n; ++i)
		scanf("%d", &sum[i]);
	for (int i = 1; i <= n; ++i)
	{
		int j;
		scanf("%d", &j);
		join(i, j);
	}
	vector<ll> vec;
	for (int i = 1; i <= n; i++)
		if (find(i), !vex[i])
			vec.push_back(sum[i]);
	sort(vec.begin(), vec.end(), greater<int>());
	ll res = 0;
	for (int i = 0; i < vec.size() && m; ++i)
		res += vec[i], m--;
	cout << res << endl;

	return 0;
}