题目链接C. Rumor
先是并查集的有关操作【模版】并查集 及路径压缩,正好可以复习一波
题意:有一个人,他所在的社区有n(1 <= n <= 1e5)个人,现在他想散播一个谣言,让整个社区的人都知道,给每个人散播谣言都要花费一定的钱,但是给一个人散播谣言之后,这个人可以免费地把谣言传给这个人的朋友。
输入:n,m(1 <= n <= 1e5)下面一行输入n个数代表散播个这个人谣言需要花费的钱,再下面m行每行两个数代表这两个人互为朋友。
输出:让所有人知道这个谣言所花费的最少的钱。
样例
Input
5 2
2 5 3 4 8
1 4
4 5
Output
10
Input
10 0
1 2 3 4 5 6 7 8 9 10
Output
55
Input
10 5
1 6 2 7 3 8 4 9 5 10
1 2
3 4
5 6
7 8
9 10
Output
15
思路:裸的并查集,找到每个集合里面最小的值相加,容易想到的就是暴力,两层for循环直接找到相同先祖元素里面元素的最小值,然后相加。虽然题目给了两秒时限,但是我交了一发,还是TLE了。然后看代码吧。
#include<bits/stdc++.h>
using namespace std;
long long fa[100005]; //记得开long long
long long a[100005];
long long find_fa(long long a)
{
if(a==fa[a])
{
return a;
}else
{
fa[a]=find_fa(fa[a]); //这里我们在找先祖元素的时候可以顺带着进行压缩操作。
return fa[a];
}
}
int main()
{
long long i,m,n,s=0,x,y;
scanf("%lld %lld",&n,&m);
for(i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
}
for(i = 1;i <= n;i++)
{
fa[i]=i; //初始化
}
for(i=1;i<=m;i++)
{
scanf("%lld %lld",&x,&y);
x=find_fa(x); //常规合并操作
y=find_fa(y);
fa[x]=y;
}
for(i=1;i<=n;i++)
{
a[find_fa(i)]=min(a[find_fa(i)],a[i]); //这里是比较巧妙的一个点,我们把所有先祖元素
} //赋值为他所在集合的最小值,就是相当于我们把每个集合的最小值都放在了先祖元素的位置上
for(i=1;i<=n;i++)
{
if(i==find_fa(i)) //然后把所有先祖元素加到一起就OK了。
{
s+=a[i];
}
}
printf("%lld\n",s);
return 0;
}