题目链接:https://ac.nowcoder.com/acm/contest/373/C
题目大意:告诉你n个点上初始逃犯的数量,再告诉你这个点的逃犯下一个时刻会逃亡的地点,相当于这两个点之间有连线,这就可以看成几个不连通的连通块,我们可以知道经过无穷个时刻之后每个逃犯都会经过自己这个联通块的某个点,所以我们可以用并查集来做,将每个连通块的点的父节点统一,将逃犯数量累加到这个父节点上,然后根据题目要求,排出最大的m个联通块的,累加就是答案。
代码:
import java.util.ArrayList;
import java.util.Collections;
import java.util.Scanner;
//图结构,用并查集查找最多的m个点的值最高的
public class C {
static int n,m;
static int maxn=100005;
static int p[]=new int[maxn];
static int find(int x){
return x==p[x]?x:(p[x]=find(p[x])); //求每个点的父节点
}
static void merge(int a,int b){
int x=find(a);
int y=find(b);
if(x!=y){ //如果y不在x这个树中,就把y加入x中
p[y]=x;
sum[x]+=sum[y]; //累加这个父节点的逃犯数量
}
}
static int a[]=new int[maxn]; //初始逃犯数量
static int v[]=new int[maxn]; //下一个时刻逃犯的地点
static long sum[]=new long[maxn]; //用来计数这个点逃犯的最后数量
public static void main(String[] args) {
Scanner sc=new Scanner(System.in);
n=sc.nextInt();
m=sc.nextInt();
for (int i = 1; i <=n ; i++) {
a[i]=sc.nextInt();
sum[i]=a[i];
p[i]=i;
}
for (int i = 1; i <=n ; i++) {
v[i]=sc.nextInt();
merge(i,v[i]);
}
ArrayList<Long> arr=new ArrayList<>();
for (int i = 1; i <=n ; i++) {
if (i==find(i)){ //将每个连通块的逃犯数量加入数组进行排序
arr.add(sum[i]);
}
}
Collections.sort(arr);
long ans=0;
for (int i = arr.size()-1; i>=0&&m>0 ; i--,m--) {
ans+=arr.get(i);
}
System.out.println(ans);
}
}