题目大意:
题目链接:https://www.luogu.org/problemnew/show/P2700
给出一棵树和几个标记点,每条边有权值,求割掉其中几条边,使得这几个标记点互不相连。求割掉的边的最小代价和。
思路:
蒟蒻忘了
K
r
u
s
k
a
l
Kruskal
Kruskal,不知道这种方法是不是
K
r
u
s
k
a
l
Kruskal
Kruskal。
我们要割掉边权和最小的几条边其实就是保留最大的边权和的边。
p
[
i
]
p[i]
p[i]表示第
i
i
i个点是否是标记点。
然后读入的每条边都用单向边储存。将边全部按权值大到小排序。
枚举每条边。设这条边的两个节点分别是
u
u
u和
v
v
v,那么如果
p
[
f
i
n
d
(
u
)
]
!
=
1
∣
∣
p
[
f
i
n
d
(
v
)
]
!
=
1
(
其
实
就
是
!
(
p
[
f
i
n
d
(
u
)
]
&
p
[
f
i
n
d
(
v
)
]
)
)
p[find(u)]!=1||p[find(v)]!=1(其实就是!(p[find(u)]\&p[find(v)]))
p[find(u)]!=1∣∣p[find(v)]!=1(其实就是!(p[find(u)]&p[find(v)])),那么就可以连这条边。那么这两个集合就合二为一,如果原来第二个集合是有标记点的,那么第一个集合的父亲就肯定要标为有标记点。即
p
[
f
i
n
d
(
u
)
]
∣
=
p
[
f
i
n
d
(
v
)
]
p[find(u)]|=p[find(v)]
p[find(u)]∣=p[find(v)]
然后
a
n
s
ans
ans加上这条边的权值即可。
最终答案就是总权值
−
a
n
s
-ans
−ans
代码:
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#define N 100100
#define ll long long
using namespace std;
int n,m,father[N],head[N],tot;
bool p[N];
ll ans,sum;
struct edge
{
int next,to,c,from;
}e[N];
void add(int from,int to,int c)
{
e[++tot].to=to;
e[tot].from=from;
e[tot].c=c;
e[tot].next=head[from];
head[from]=tot;
}
bool cmp(edge x,edge y)
{
return x.c>y.c;
}
int find(int x) //寻找祖先(路径压缩)
{
return x==father[x]?x:father[x]=find(father[x]);
}
int main()
{
scanf("%d%d",&n,&m);
int x,y,z;
for (int i=1;i<=m;i++)
{
scanf("%d",&x);
p[x+1]=1; //标记点
}
for (int i=1;i<n;i++)
{
scanf("%d%d%d",&x,&y,&z);
add(x+1,y+1,z);
sum+=(ll)z; //总权值
}
sort(e+1,e+n,cmp);
for (int i=1;i<=n;i++)
father[i]=i;
for (int i=1;i<n;i++)
if (!(p[find(e[i].from)]&p[find(e[i].to)])) //不能两边都是标记点
{
ans+=(ll)e[i].c;
p[find(e[i].from)]|=p[find(e[i].to)];
father[find(e[i].to)]=father[find(e[i].from)]; //合二为一
}
cout<<sum-ans;
return 0;
}