数据结构专题系列 基础篇-8 最小生成树
--------------------------------------------------------------------------------
题目传送门
洛谷最小生成树 普及-组
洛谷最小生成树 普及/提高-组
--------------------------------------------------------------------------------
更多详见>>
OJ题解系列 目录导航帖
--------------------------------------------------------------------------------
序
这里是数据结构专题系列 基础篇-8 最小生成树
最小生成树作为树的一个分支,在简化图的算法中有着很好的应用,能够求解出连通图的最小边权和,同样,最小生成树会有许多新的变式,在基础部分我们只介绍最大生成树这一个扩展应用
基础部分的题目都是最小生成树的模板题(很多题目代码都基本一致,只是读入输出略有差别),第一道[模板]题,我们采用Prim算法加以解决,而剩余的其他题目我们则采用Kruskal算法加以解决,两个算法的复杂度分别为
O
(
V
l
o
g
V
+
E
)
,
O
(
(
V
+
E
)
l
o
g
V
)
O(VlogV+E),O((V+E)logV)
O(VlogV+E),O((V+E)logV)
接下来就是题解部分了,每道算法题都标注有对应的算法标签,对于那些易错、较难或是测试点比较特殊的题目会着重标注,本章推荐的题目有:
P2916 [USACO08NOV]Cheering up the Cow G | 树,二叉树 + 最小生成树(变式) |
---|---|
P4826 [USACO15FEB]Superbull S | 树,二叉树 + 最小生成树(最大生成树) |
其余题目 | 模板题(适合练习) |
--------------------------------------------------------------------------------
难度: 普及-
P3366 【模板】最小生成树
算法标签: 树,二叉树 + 最小生成树
注意点: 模板题,这里用Prim算法实现
#include<bits/stdc++.h>
using namespace std;
const int INF = 0x3f3f3f3f;
int g[5005][5005];
int vis[5005];
int d[5005];
long long ans = 0;
int N,M;
void Prim(int v0){
int i,j,k,minn;
memset(vis,0,sizeof(vis));
for(int i=1;i<=N;i++){
d[i]=INF;
}
d[v0]=0;
ans = 0;
for(int i=1;i<=N;i++){
minn = INF;
for(int j=1;j<=N;j++){
if(vis[j]==0 && minn>d[j]){
minn = d[j];
k=j;
}
}
vis[k]=1;
ans+=d[k];
for(int j=1;j<=N;j++){
if(vis[j]==0 && d[j]>g[k][j]){
d[j]=g[k][j];
}
}
}
}
int main(){
cin>> N >>M;
for(int i=1;i<=N;i++){
for(int j=1;j<=N;j++){
g[i][j]=INF;
}
}
for(int i=0;i<M;i++){
int x,y,z;
cin >>x >>y >> z;
g[x][y]=min(g[x][y],z);
g[y][x]=min(g[y][x],z);
}
Prim(1);
cout << ans <<endl;
return 0;
}
--------------------------------------------------------------------------------
难度: 普及/提高-
P2916 [USACO08NOV]Cheering up the Cow G
算法标签: 树,二叉树 + 最小生成树
注意点: 一道好题!考察了最小生成树的变形结构(本题采用kruskal算法):
由题意知,每条边和每个顶点至少被访问两次,因此最小生成树的边权=2个点权+边权*2,在读入数据的时候,就要对边权进行更新,此外,我们还需要选定一个起始顶点,这个点如果是两端的端点,那么不需要再计算一次(出发,返回,一共2次),但如果是内部顶点,还需计算一次(出发+回溯+返回,一共3次)
#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e4+5;
int a[maxn];
struct node{
int u,v;
int dis;
};
node road[10*maxn];
bool cmp(node r1,node r2){
return r1.dis < r2.dis;
}
int father[maxn];
int findfather(int x){
if(x == father[x]){
return x;
}
int F = findfather(father[x]);
father[x] = F;
return F;
}
void Union(int x,int y){
int fx = findfather(x);
int fy = findfather(y);
if(fx!=fy){
father[fx] = fy;
}
}
int main(){
int N,P;
scanf("%d%d",&N,&P);
int minn = INT_MAX;
for(int i=1;i<=N;i++){
scanf("%d",&a[i]);
minn = min(minn,a[i]);
}
for(int i=0;i<P;i++){
scanf("%d%d%d",&road[i].u,&road[i].v,&road[i].dis);
road[i].dis *= 2;
road[i].dis += a[road[i].u];
road[i].dis += a[road[i].v];
}
sort(road,road+P,cmp);
for(int i=1;i<=N;i++){
father[i] = i;
}
int cnt = 0;
int ans = 0;
for(int i=0;i<P && cnt<N-1;i++){
int u = road[i].u;
int v = road[i].v;
if(findfather(u)!=findfather(v)){
Union(u,v);
cnt++;
ans += road[i].dis;
}
}
cout << ans+minn << endl;
return 0;
}
P2330 [SCOI2005]繁忙的都市
算法标签: 树,二叉树 + 最小生成树
注意点: 模板题 + kruskal
#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5+5;
int n,m,s,t;
int prt[maxn];
typedef struct xiaoming{
int x;
int y;
int z;
}xiaoming;
xiaoming a[maxn];
bool cmp(const xiaoming &x1,const xiaoming &x2){
return x1.z<x2.z;
}
int Getfather(int x){
if(prt[x]==x){
return x;
}
prt[x] = Getfather(prt[x]);
return prt[x];
}
void Kruskal(){
int f1,f2,k;
k=0;
for(int i=1;i<=n;i++){
prt[i]=i;
}
for(int i=1;i<=m;i++){
f1 = Getfather(a[i].x);
f2 = Getfather(a[i].y);
if(f1!=f2){
prt[f1] = f2;
k++;
if(k==n-1){
cout << a[i].z << endl;
break;
}
}
}
}
int main(){
cin >> n >> m ;
for(int i=1;i<=m;i++){
cin >> a[i].x >> a[i].y >> a[i].z;
}
cout << n-1 << " ";
sort(a+1,a+m+1,cmp);
Kruskal();
return 0;
}
P1396 营救
算法标签: 树,二叉树 + 最小生成树
注意点: 模板题 + kruskal
#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e4*2+5;
int n,m,s,t;
int prt[maxn];
typedef struct xiaoming{
int x;
int y;
int z;
}xiaoming;
xiaoming a[maxn];
bool cmp(const xiaoming &x1,const xiaoming &x2){
return x1.z<x2.z;
}
int Getfather(int x){
if(prt[x]==x){
return x;
}
prt[x] = Getfather(prt[x]);
return prt[x];
}
void Kruskal(){
int f1,f2,k;
k=0;
for(int i=1;i<=n;i++){
prt[i]=i;
}
for(int i=1;i<=m;i++){
f1 = Getfather(a[i].x);
f2 = Getfather(a[i].y);
if(f1!=f2){
prt[f1] = f2;
k++;
if(Getfather(s)==Getfather(t)){
cout << a[i].z << endl;
break;
}
}
}
}
int main(){
cin >> n >> m >> s >> t;
for(int i=1;i<=m;i++){
cin >> a[i].x >> a[i].y >> a[i].z;
}
sort(a+1,a+m+1,cmp);
Kruskal();
return 0;
}
P1546 [USACO3.1]最短网络 Agri-Net
算法标签: 树,二叉树 + 最小生成树
注意点: 模板题 + kruskal
#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5+5;
int n,m=0;
long long ans = 0;
int t[105][105];
int prt[maxn];
typedef struct xiaoming{
int x;
int y;
int z;
}xiaoming;
xiaoming a[maxn];
bool cmp(const xiaoming &x1,const xiaoming &x2){
return x1.z<x2.z;
}
int Getfather(int x){
if(prt[x]==x){
return x;
}
prt[x] = Getfather(prt[x]);
return prt[x];
}
void Kruskal(){
int f1,f2,k;
k=0;
for(int i=1;i<=n;i++){
prt[i]=i;
}
for(int i=1;i<=m;i++){
f1 = Getfather(a[i].x);
f2 = Getfather(a[i].y);
if(f1!=f2){
ans = ans +a[i].z;
prt[f1] = f2;
k++;
if(k==n-1){
cout << ans << endl;
break;
}
}
}
}
int main(){
cin >> n;
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
cin >> t[i][j];
}
}
for(int i=1;i<=n-1;i++){
for(int j=i+1;j<=n;j++){
++m;
a[m].x=i;
a[m].y=j;
a[m].z=t[i][j];
}
}
sort(a+1,a+m+1,cmp);
Kruskal();
return 0;
}
P2872 [USACO07DEC]Building Roads S
算法标签: 树,二叉树 + 最小生成树
注意点: 模板题 + kruskal
#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e6+1e3+5;
int prt[maxn];
double ans = 0.0;
typedef struct farm{
int x;
int y;
double z;
}farm;
farm a[maxn];
int N,M;
int sum = 1;
bool cmp(const farm &x1,const farm &y1){
return x1.z<y1.z;
}
typedef struct city{
int x;
int y;
}city;
city c[maxn];
double dis(int x,int y){
return (double)sqrt((double)(c[x].x-c[y].x)*(c[x].x-c[y].x)+(double)(c[x].y-c[y].y)*(c[x].y-c[y].y));
}
int Getfather(int x){
if(prt[x]==x){
return x;
}
prt[x] = Getfather(prt[x]);
return prt[x];
}
void Kruskal(){
int f1,f2,k;
k=0;
for(int i=1;i<=N;i++){
prt[i]=i;
}
for(int i=1;i<sum;i++){
f1 = Getfather(a[i].x);
f2 = Getfather(a[i].y);
if(f1!=f2){
ans = ans + a[i].z;
prt[f1] = f2;
k++;
if(k==N-1){
break;
}
}
}
}
int main(){
cin >> N >> M;
for(int i=1;i<=N;i++){
cin>> c[i].x >> c[i].y;
}
for(int i=1;i<N;i++){
for(int j=i+1;j<=N;j++){
a[sum].x = i;
a[sum].y = j;
a[sum++].z = dis(i,j);
}
}
for(int i=sum;i<sum+M;i++){
cin >> a[i].x >> a[i].y;
a[i].z = 0;
}
sum = sum + M;
sort(a+1,a+sum,cmp);
Kruskal();
printf("%.2lf\n",ans);
return 0;
}
P1194 买礼物
算法标签: 树,二叉树 + 最小生成树
注意点: 模板题 + kruskal + 最优选取
注意题面中的这句话:特别的,如果
K
I
,
J
=
0
K_{I,J}=0
KI,J=0,那么表示这两样东西之间不会导致优惠,不要理解成可通过0元获取!
#include<bits/stdc++.h>
using namespace std;
const int maxn = 5e5+5;
int a[505][505];
int father[505];
int cnt = 0;
struct node{
int x;
int y;
int dis;
};
node t[maxn];
bool cmp(node t1,node t2){
return t1.dis < t2.dis;
}
int findfather(int x){
if(x == father[x]){
return x;
}
int F = findfather(father[x]);
father[x] = F;
return F;
}
void Union(int x,int y){
int fx = findfather(x);
int fy = findfather(y);
if(fx!=fy){
father[fx] = fy;
}
}
int main(){
int A,B;
cin >> A >> B;
for(int i=1;i<=B;i++){
father[i] = i;
}
for(int i=1;i<=B;i++){
for(int j=1;j<=B;j++){
scanf("%d",&a[i][j]);
}
}
for(int i=1;i<=B-1;i++){
for(int j=i+1;j<=B;j++){
if(a[i][j]){
t[cnt].x = i;
t[cnt].y = j;
t[cnt].dis = a[i][j];
cnt++;
}
t[cnt].x = i;
t[cnt].y = j;
t[cnt].dis = A;
cnt++;
}
}
sort(t,t+cnt,cmp);
int ans = 0;
int count = 0;
for(int i=0;i<cnt && count<B-1;i++){
int u = t[i].x;
int v = t[i].y;
if(findfather(u)!=findfather(v)){
Union(u,v);
count++;
ans += t[i].dis;
}
}
cout << ans + A << endl;
return 0;
}
P1547 [USACO05MAR]Out of Hay S
算法标签: 树,二叉树 + 最小生成树
注意点: 模板题 + kruskal
#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5+5;
int n,m,s,t;
int prt[maxn];
typedef struct xiaoming{
int x;
int y;
int z;
}xiaoming;
xiaoming a[maxn];
bool cmp(const xiaoming &x1,const xiaoming &x2){
return x1.z<x2.z;
}
int Getfather(int x){
if(prt[x]==x){
return x;
}
prt[x] = Getfather(prt[x]);
return prt[x];
}
void Kruskal(){
int f1,f2,k;
k=0;
for(int i=1;i<=n;i++){
prt[i]=i;
}
for(int i=1;i<=m;i++){
f1 = Getfather(a[i].x);
f2 = Getfather(a[i].y);
if(f1!=f2){
prt[f1] = f2;
k++;
if(k==n-1){
cout << a[i].z << endl;
break;
}
}
}
}
int main(){
cin >> n >> m ;
for(int i=1;i<=m;i++){
cin >> a[i].x >> a[i].y >> a[i].z;
}
sort(a+1,a+m+1,cmp);
Kruskal();
return 0;
}
P2820 局域网
算法标签: 树,二叉树 + 最小生成树
注意点: 模板题 + kruskal
#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5+5;
int n,m,s,t;
long long ans = 0;
long long sum = 0;
int prt[maxn];
typedef struct xiaoming{
int x;
int y;
int z;
}xiaoming;
xiaoming a[maxn];
bool cmp(const xiaoming &x1,const xiaoming &x2){
return x1.z<x2.z;
}
int Getfather(int x){
if(prt[x]==x){
return x;
}
prt[x] = Getfather(prt[x]);
return prt[x];
}
void Kruskal(){
int f1,f2,k;
k=0;
for(int i=1;i<=n;i++){
prt[i]=i;
}
for(int i=1;i<=m;i++){
f1 = Getfather(a[i].x);
f2 = Getfather(a[i].y);
if(f1!=f2){
ans = ans +a[i].z;
prt[f1] = f2;
k++;
if(k==n-1){
cout << sum-ans << endl;
break;
}
}
}
}
int main(){
cin >> n >> m ;
for(int i=1;i<=m;i++){
cin >> a[i].x >> a[i].y >> a[i].z;
sum+=a[i].z;
}
sort(a+1,a+m+1,cmp);
Kruskal();
return 0;
}
P2504 [HAOI2006]聪明的猴子
算法标签: 树,二叉树 + 最小生成树
注意点: 模板题 + kruskal + 二维坐标距离
#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e6+5;
int a[505];
int father[1005];
int cnt = 0;
struct node{
int x;
int y;
double dis;
};
node t[maxn];
struct input{
int x;
int y;
};
input in[1005];
bool cmp(node t1,node t2){
return t1.dis < t2.dis;
}
int findfather(int x){
if(x == father[x]){
return x;
}
int F = findfather(father[x]);
father[x] = F;
return F;
}
void Union(int x,int y){
int fx = findfather(x);
int fy = findfather(y);
if(fx!=fy){
father[fx] = fy;
}
}
int main(){
int M;
cin >> M;
for(int i=1;i<=M;i++){
cin >> a[i];
}
int N;
cin >> N;
for(int i=1;i<=N;i++){
cin >> in[i].x >> in[i].y;
}
for(int i=1;i<=N-1;i++){
for(int j=i+1;j<=N;j++){
t[cnt].x = i;
t[cnt].y = j;
t[cnt].dis = sqrt( (in[i].x-in[j].x)*(in[i].x-in[j].x) +
(in[i].y-in[j].y)*(in[i].y-in[j].y) );
cnt++;
}
}
sort(t,t+cnt,cmp);
for(int i=1;i<=N;i++){
father[i] = i;
}
int count = 0;
double ans = 0.0;
for(int i=0;i<cnt && count<N-1;i++){
int u = t[i].x;
int v = t[i].y;
if(findfather(u) != findfather(v)){
Union(u,v);
count++;
ans = t[i].dis;
}
}
int res = 0;
for(int i=1;i<=M;i++){
if(a[i]>=ans){
res++;
}
}
cout << res << endl;
return 0;
}
P1991 无线通讯网
算法标签: 树,二叉树 + 最小生成树
注意点: 模板题 + kruskal + 二维坐标距离
#include<bits/stdc++.h>
using namespace std;
typedef struct city{
int x;
int y;
double z;
}city;
const int maxn = 1e5*2+5;
int prt[maxn];
city a[maxn];
int x[maxn];
int y[maxn];
int S,P;
double ans = 0.0;
int sum = 1;
double dis(int x1,int x2,int y1,int y2){
return sqrt((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2));
}
bool cmp(const city &x,const city &y){
return x.z < y.z;
}
int Getfather(int x){
if(prt[x]==x){
return x;
}
prt[x]=Getfather(prt[x]);
return prt[x];
}
void Kruskal(){
int f1,f2,k;
k=0;
for(int i=1;i<=P;i++){
prt[i]=i;
}
for(int i=1;i<sum;i++){
f1 = Getfather(a[i].x);
f2 = Getfather(a[i].y);
if(f1!=f2){
ans = a[i].z;
k++;
prt[f1]=f2;
if(k==P-S){
break;
}
}
}
}
int main(){
cin >> S >> P;
for(int i=1;i<=P;i++){
cin >> x[i] >> y[i];
}
for(int i=1;i<=P-1;i++){
for(int j=i+1;j<=P;j++){
a[sum].x=i;
a[sum].y=j;
a[sum].z = dis(x[i],x[j],y[i],y[j]);
sum++;
}
}
sort(a+1,a+sum+1,cmp);
Kruskal();
printf("%.2lf\n",ans);
return 0;
}
P4826 [USACO15FEB]Superbull S
算法标签: 树,二叉树 + 最大生成树
注意点: 模板题 + kruskal (最大生成树)
#include<bits/stdc++.h>
using namespace std;
const int maxn = 4e6+5;
int a[2005];
int father[2005];
int cnt = 0;
struct node{
int x;
int y;
int dis;
};
node t[maxn];
bool cmp(node t1,node t2){
return t1.dis > t2.dis;
}
int findfather(int x){
if(x == father[x]){
return x;
}
int F = findfather(father[x]);
father[x] = F;
return F;
}
void Union(int x,int y){
int fx = findfather(x);
int fy = findfather(y);
if(fx!=fy){
father[fx] = fy;
}
}
int main(){
int n;
cin >> n;
for(int i=1;i<=n;i++){
cin >> a[i];
}
for(int i=1;i<=n-1;i++){
for(int j=i+1;j<=n;j++){
t[cnt].x = i;
t[cnt].y = j;
t[cnt].dis = a[i] ^ a[j];
cnt++;
}
}
for(int i=1;i<=n;i++){
father[i] = i;
}
sort(t,t+cnt,cmp);
long long ans = 0;
int count = 0;
for(int i=0;i<cnt && count<n-1;i++){
int u = t[i].x;
int v = t[i].y;
if(findfather(u)!=findfather(v)){
Union(u,v);
count++;
ans += t[i].dis;
}
}
cout << ans << endl;
return 0;
}
P1195 口袋的天空
算法标签: 树,二叉树 + 最小生成树
注意点: 模板题 + kruskal
#include<bits/stdc++.h>
using namespace std;
int prt[1005];
int N,M,K;
typedef struct Edge{
int x;
int y;
int z;
}Edge;
Edge a[10005];
bool cmp(const Edge &x,const Edge &y){
return x.z< y.z;
}
long long ans = 0;
int Getfather(int x){
if(prt[x]==x){
return x;
}
prt[x]=Getfather(prt[x]);
return prt[x];
}
void Kruskal(){
int f1,f2,k;
k=0;
for(int i=1;i<=N;i++){
prt[i]=i;
}
for(int i=1;i<=M;i++){
f1 = Getfather(a[i].x);
f2 = Getfather(a[i].y);
if(f1!=f2){
ans = ans + a[i].z;
prt[f1] = f2;
k++;
if(k==N-K){
break;
}
}
}
}
int main(){
cin >> N >> M >> K;
if(N-M>K){
cout << "No Answer"<<endl;
return 0;
}
for(int i=1;i<=M;i++){
int x,y,z;
cin >> x >> y >> z;
a[i].x = x;
a[i].y = y;
a[i].z = z;
}
sort(a+1,a+M+1,cmp);
Kruskal();
cout << ans << endl;
return 0;
}