最小生成树模板(kruskal&&prim)

1.kruskal

  • 最小生成树能够保证首先是树(对于n个顶点的图只有条n - 1边),其次保证任意两个顶点之间都可达,再次保证这棵树的边权值之和为最小,但不能保证任意两点之间是最短路径;(最后求出来的路径长度是经过n个顶点的)
  • 最小生成树是到一群点(所有点)的路径代价和最小,是一个n - 1 条边的树,最短路是从一个点到另一个点的最短路径;

求最大生成树(所有点的路径代价之和最大)即把边从大到小排序

#include <bits/stdc++.h>
using namespace std;
const int N=1e3;
const int inf=0x3f3f3f3f;

struct node
{
    int u,v,w;
    bool operator <(const node &a)const
    {
        return w<a.w;
    }
}edge[N];

int father[N];
int n,m;

void init()
{
    for(int i=0;i<m;i++)
        father[i]=i;
}

int Find(int x)
{
    if(x==father[x])
        return x;
    return father[x]=Find(father[x]);
}

void Union(int x,int y)
{
    int tmpx=Find(x);
    int tmpy=Find(y);
    if(tmpx!=tmpy)
    {
        father[tmpx]=tmpy;
    }
}

int kruskal()
{
    sort(edge,edge+n);
    init();
    node now;
    int ans=0;
    for(int i=0;i<n;i++)
    {
        now=edge[i];
        if(Find(now.u)!=Find(now.v))
        {
            Union(now.u,now.v);
            ans+=now.w;
        }
    }
    return ans;
}

int main()
{
    while(scanf("%d%d",&n,&m)!=EOF&&n)
    {
        for(int i=0;i<n;++i)
        {
            scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
        }
        int ans=kruskal();
        bool f=0;
        for(int i=2;i<=m;i++)
        {
            if(Find(1)!=Find(i))
            {
                f=1;
                break;
            }
        }
        if(f)
            cout<<"?"<<'\n';
        else
            cout<<ans<<'\n';
    }
    return 0;
}

2.prim

#include <bits/stdc++.h>
using namespace std;
const int N=1e3;
const int inf=0x3f3f3f3f;
int n,m;
int dis[N],vis[N],g[N][N];
int prim(int src)
{
    for(int i=1;i<=m;i++)
    {
        dis[i]=g[src][i];
    }
    int sum=0;
    vis[src]=1;
    dis[src]=0;
    for(int i=2;i<=m;i++)
    {
        int pos=-1;
        int minn=inf;
        for(int j=1;j<=m;j++)
        {
            if(dis[j]<minn&&!vis[j])
            {
                pos=j;
                minn=dis[j];
            }
        }
        if(pos==-1)
        {
            return -1;
        }
        sum+=minn;
        vis[pos]=1;
        for(int j=1;j<=m;j++)
        {
            if(!vis[j]&&dis[j]>g[pos][j])
            {
                dis[j]=g[pos][j];
            }
        }
    }
    return sum;
}
int main()
{
    int a,b,c;
    while(scanf("%d%d",&n,&m)!=EOF&&n)
    {
        memset(g,inf,sizeof(g));
        memset(dis,inf,sizeof(dis));
        memset(vis,0,sizeof(vis));
        for(int i=0;i<n;i++)
        {
            scanf("%d%d%d",&a,&b,&c);
            if(c<g[a][b])
                g[a][b]=g[b][a]=c;
        }
        int ans=prim(1);
        if(ans==-1)
            cout<<"?"<<'\n';
        else
            cout<<ans<<'\n';
    }
    return 0;
}

 

全部评论

相关推荐

ProMonkey2024:5个oc?厉害! 但是有一个小问题:谁问你了?😡我的意思是,谁在意?我告诉你,根本没人问你,在我们之中0人问了你,我把所有问你的人都请来 party 了,到场人数是0个人,誰问你了?WHO ASKED?谁问汝矣?誰があなたに聞きましたか?누가 물어봤어?我爬上了珠穆朗玛峰也没找到谁问你了,我刚刚潜入了世界上最大的射电望远镜也没开到那个问你的人的盒,在找到谁问你之前我连癌症的解药都发明了出来,我开了最大距离渲染也没找到谁问你了我活在这个被辐射蹂躏了多年的破碎世界的坟墓里目睹全球核战争把人类文明毁灭也没见到谁问你了(别的帖子偷来的,现学现卖😋)
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务