最小生成树相关（Kruskal,Prim,附拓扑排序）（未完成）

C_eeking

已于 2023-01-22 20:02:43 修改

阅读量485

点赞数

分类专栏：图论文章标签： c++ 数据结构算法

于 2022-04-22 18:22:15 首次发布

本文链接：https://blog.csdn.net/C_eeking/article/details/121040486

版权

图论专栏收录该内容

12 篇文章 0 订阅

订阅专栏

最小生成树相关（Kruskal,Prim,附拓扑排序）

算法思想
总结
参考文献

算法思想

Kruskal

Kruskal是解决最小生成树的常用算法（这里的小根据题目的定义来，可能是距离或者权值等），其基本思路是边上的贪心，每次选择未被收录的长度最小的边，将其加入集合，直到无边可选，设边数为M，点数为N，Kruskal的时间复杂度为 $O(M\log M)$ ，也就说，不适宜点多的情况

代码

void Kruskal() {
    while(!q.empty()) {//使用的是以边权为基准的优先队列
        node t=q.top();
        q.pop();
        if(Seek(t.u)==Seek(t.v))continue;//如果是一个祖先
        Union(t.u,t.v);
        ans+=t.w;
        cnt++;
        if(cnt==n-1)break;//特判避免不必要的计算
    }
}

题目举例（洛谷P3366）

#include <bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
const int maxn=2e5+10;
const int N=5001;
using namespace std;
struct node {
    int u,v,w;
    bool operator<(const node t)const {
        return w>t.w;
    }
};
int f[N],n,m,ans,cnt;
priority_queue<node,vector<node>,less<node>>q;
int Seek(int x) {//并查集路径压缩
    return x==f[x]?x:f[x]=Seek(f[x]);
}
void Union(int x,int y) {//并查集合并
    int fx=Seek(x),fy=Seek(y);
    if(fx!=fy)f[fx]=fy;
}
void Kruskal() {
    while(!q.empty()) {
        node t=q.top();
        q.pop();
        if(Seek(t.u)==Seek(t.v))continue;
        Union(t.u,t.v);
        ans+=t.w;
        cnt++;
        if(cnt==n-1)break;
    }
}
int main() {
    ios::sync_with_stdio();
    cin.tie(0);
    cin >>n>>m;
    for(int i=1; i<=n; i++)f[i]=i;
    for(int i=1; i<=m; i++) {
        int u,v,w;
        cin >>u>>v>>w;
        q.push({u,v,w});
    }
    Kruskal();
    cnt==n-1?cout <<ans:cout<<"orz";
    return 0;
}

Prim

与Kruscal相反，Prim算法的主要操作对象是点，算法思想也是贪心，基本思路是选择到当前已构造的生成树的最小距离的点，将其收录，用该点更新其相邻节点到生成树的距离

代码

void Prim() {
    memset(dis,0x3f,sizeof(dis));//初始化无穷大
    priority_queue<pr,vector<pr>,greater<pr>>q;//以树距为基准的优先队列
    q.push({0,1});
    dis[1]=0;//初始化起点
    while(!q.empty()) {
        pr t=q.top();
        q.pop();
        int u=t.second;
        if(vis[u])continue;
        vis[u]=1;
        ans+=dis[u];//累和记录
        acc++;//记录已经收录节点数量
        for(int i=head[u]; i; i=e[i].next) {
            int v=e[i].to;
            if(vis[v])continue;
            if(dis[v]>e[i].w) {//如果可以松弛
                q.push({e[i].w,v});
                dis[v]=e[i].w;
            }
        }
        if(acc==n)break;
    }
}

题目举例（PTA公路村村通）

#include <bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
const int maxn=1e5+10;
const int N=1001;
using namespace std;
typedef pair<int,int>pr;
int n,m,head[N],cnt,dis[N],ans,acc;
bool vis[N];
struct node {
    int next,to,w;
} e[maxn];
void Add(int from,int to,int w) {
    e[++cnt].next=head[from];
    e[cnt].to=to;
    e[cnt].w=w;
    head[from]=cnt;
}
void Prim() {
    memset(dis,0x3f,sizeof(dis));
    priority_queue<pr,vector<pr>,greater<pr>>q;
    q.push({0,1});
    dis[1]=0;
    while(!q.empty()) {
        pr t=q.top();
        q.pop();
        int u=t.second;
        if(vis[u])continue;
        vis[u]=1;
        ans+=dis[u];
        acc++;
        for(int i=head[u]; i; i=e[i].next) {
            int v=e[i].to;
            if(vis[v])continue;
            if(dis[v]>e[i].w) {
                q.push({e[i].w,v});
                dis[v]=e[i].w;
            }
        }
        if(acc==n)break;
    }
}
int main() {
    ios::sync_with_stdio();
    cin.tie(0);
    cin >>n>>m;
    while(m--) {
        int u,v,w;
        cin >>u>>v>>w;
        Add(u,v,w);
        Add(v,u,w);
    }
    Prim();
    acc==n?cout <<ans:cout <<-1;
    return 0;
}

拓扑排序

拓扑排序是一个较简单的知识点，基本要求是：在一个有向图中，对所有节点进行排序，要求没有一个节点指向它前面的节点

一般的处理步骤：统计节点入度，对所有入度为0节点按照给定的规则（例如字典序）抽取并更新各自的邻接点的入度，以此类推，如果最后不存在入度为0的节点，代表存在环，无解

代码

 while(k--)
      if(!q.empty()) {
          int len=q.size();
          n-=len;
          while(len--) {
              int u=q.front();
              q.pop();
              for(int i=head[u]; i; i=e[i].next) {
                  int v=e[i].to;
                  in[v]--;
                  if(in[v]==0)
                      q.push(v);
              }
          }
      }

题目举例（Codeforces1593E）

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=8e5+10;
int n,t,head[maxn>>1],cnt,k,in[maxn>>1];
struct node {
    int to,next;
} e[maxn];
void Add(int from,int to) {
    e[++cnt].to=to;
    e[cnt].next=head[from];
    head[from]=cnt;
}
void solve() {
    scanf("%d%d",&n,&k);
    for(int i=1; i<n; i++) {
        int u,v;
        scanf("%d%d",&u,&v);
        Add(u,v);
        Add(v,u);
        in[u]++,in[v]++;
    }
    queue<int>q;
    for(int i=1; i<=n; i++)if(in[i]==1)q.push(i);
    if(k>=n) {
        printf("0\n");
        cnt=0;
        memset(head,0,sizeof(int)*(n+1));
        memset(in,0,sizeof(int)*(n+1));
        return;
    }
    while(k--)
        if(!q.empty()) {
            int len=q.size();
            n-=len;
            while(len--) {
                int u=q.front();
                q.pop();
                in[u]=0;
                for(int i=head[u]; i; i=e[i].next) {
                    int v=e[i].to;
                    in[v]--;
                    if(in[v]==1)
                        q.push(v);
                }
            }
        }
    printf("%d\n",n<=0?0:n);
    cnt=0;
    memset(head,0,sizeof(head));
    memset(in,0,sizeof(in));
}
int main() {
    scanf("%d",&t);
    while(t--)
        solve();
    return 0;
}

训练

POJ1789

题目大意：给出一张无向带权图，求出最小生成树的权值和，但建边和边权题目自有定义

思路：基本上是模板题，只需要注意如何建图即可，由于给出的数据不多，所以可以考虑直接暴力建图，每个点与其他点的字符串相互比较得到边权（poj什么时候能更新编译器啊）

代码

//#include <bits/stdc++.h>
#include <cstdlib>
#include <queue>
#include <cstdio>
#include <string>
#include <cstring>
#include <iostream>
//#define int long long
#include <iterator>
using namespace std;
const int maxn=5e6+100;
int n,cnt,head[2121],dis[2121];
bool vis[2121];
typedef pair<int,int>pr;
struct node {
    int next,to,w;
    bool operator<(const node t)const {
        return w>t.w;
    }
} e[maxn];
char point[2121][10];
void Add(int from,int to,int w) {
    e[++cnt].next=head[from];
    e[cnt].to=to;
    e[cnt].w=w;
    head[from]=cnt;
}
int judge(int i,int j) {//判断边权
    int ans=0;
    for(int p=0; p<7; p++)
        if(point[i][p]!=point[j][p])ans++;
    return ans;
}
int Prim() {
    memset(dis,0x3f,sizeof(dis));
    int ans=0,acc=0;
    priority_queue<pr,vector<pr>,greater<pr> >q;
    pr t;
    t.first=0,t.second=1;
    q.push(t);
    dis[1]=0;
    while(!q.empty()) {
        pr t=q.top();
        q.pop();
        int u=t.second;
        if(vis[u])continue;
        vis[u]=1;
        ans+=dis[u];
        acc++;
        for(int i=head[u]; i; i=e[i].next) {
            int v=e[i].to;
            if(vis[v])continue;
            if(dis[v]>e[i].w) {
                pr tmp;
                tmp.first=e[i].w,tmp.second=v;
                q.push(tmp);
                dis[v]=e[i].w;
            }
        }
        if(acc==n)break;
    }
    return ans;
}

int main() {
//    ios::sync_with_stdio(0);
//    cin.tie(0);
    while(scanf("%d",&n)&&n) {
        for(int i=1; i<=n; ++i) {
            scanf("%s",point[i]);
            vis[i]=0,head[i]=0;
        }
        cnt=0;
        for(int i=1; i<=n; ++i)
            for(int j=i+1; j<=n; ++j) {
                Add(i,j,judge(i,j));//建图
                Add(j,i,judge(i,j));
            }
        printf("The highest possible quality is 1/%d.\n",Prim());
    }
    return 0;
}
/*
4
aaaaaaa
baaaaaa
abaaaaa
aabaaaa
0
*/

POJ2485

题目大意：

思路：

代码

#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <queue>
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;
const int maxn=1e6;
int head[maxn],cnt,T,n,dis[maxn];
bool vis[maxn];
typedef pair<int,int>pr;
struct node {
    int next,to,w;
} e[maxn];
void Add(int from,int to,int w) {
    e[++cnt].next=head[from];
    e[cnt].to=to;
    e[cnt].w=w;
    head[from]=cnt;
}
int Prim() {
    priority_queue<pr,vector<pr>,greater<pr> >q;
    int acc=0,res=0;
    q.push({0,1});
    dis[1]=0;
    while(!q.empty()) {
        pr t=q.top();
        q.pop();
        int u=t.second;
        if(vis[u])continue;
        vis[u]=1;
        res=max(res,dis[u]);
        acc++;
        for(int i=head[u]; ~i; i=e[i].next) {
            int v=e[i].to;
            if(vis[v])continue;
            if(dis[v]>e[i].w) {
                q.push({e[i].w,v});
                dis[v]=e[i].w;
            }
        }
        if(acc==n)break;
    }
    return res;
}
int main() {
    scanf("%d",&T);
    while(T--) {
        memset(head,-1,sizeof(head));
        memset(dis,0x3f,sizeof(dis));
        memset(vis,0,sizeof(vis));
        cnt=0;
        scanf("%d",&n);
        for(int i=1; i<=n; i++)
            for(int j=1; j<=n; j++) {
                int x;
                scanf("%d",&x);
                if(x==0)continue;
                Add(i,j,x);
            }
        printf("%d\n",Prim());
    }
    return 0;
}

POJ3026

题目大意：

思路：

代码

POJ1094

题目大意：

思路：

代码

PTA关键活动

题目大意：

思路：

代码

洛谷P1967

题目大意：

思路：

代码

洛谷P4180

题目大意：

思路：

代码

总结

参考文献

《啊哈算法》

C_eeking

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
1
评论
最小生成树相关（Kruskal,Prim,附拓扑排序）（未完成）

最小生成树相关（Kruskal,Prim,附拓扑排序）算法思想KruskalPrim拓扑排序训练POJ1789POJ2485POJ1258POJ3026POJ1094PTA关键活动洛谷P1967洛谷P4180总结参考文献算法思想KruskalKruskal是解决最小生成树的常用算法（这里的小根据题目的定义来，可能是距离或者权值等）代码题目举例（洛谷P3366）Prim代码题目举例（PTA公路村村通）拓扑排序代码题目举例（Codeforces1593E）训练POJ1789
复制链接

扫一扫

专栏目录