图论模板——kruscal,dijkstra

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e5+10;
int fi[maxn],ne[maxn],v[maxn],w[maxn];
struct re
{
    int x,y,z;
};
re a[maxn];
int fa[maxn];
int t=-1;

bool cmp(re a,re b)
{
    return a.z<b.z;
}
int find(int k)
{
    if (k==fa[k]) return k;
    else return find(fa[k]);
}
int main()
{
    int n,m;
    cin>>n>>m;
    for (int i=1;i<=n;i++) fa[i]=i;
    for (int i=1;i<=m;i++)
    {       
        scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].z);
    }
    sort(a+1,a+1+m,cmp);    
    int ans=0;
    int mark=0;
    for (int i=1;i<=m;i++)
    {
        if (find(a[i].x)==find(a[i].y)) continue;
        ans+=a[i].z;
        fa[find(a[i].x)]=find(a[i].y);
        mark++;
        if (mark==n-1) break;
    }
    if (mark!=n-1) cout<<"orz";else cout<<ans;  
}

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,m,s,t;
#define inf 2147483647
#define maxn 501000
int dis[maxn];
bool f[maxn];
int fi[maxn],ne[maxn],u[maxn],v[maxn],w[maxn];
void edge(int x,int y,int z)
{
    t++;
    ne[t]=fi[x];
    fi[x]=t;
    v[t]=y;
    w[t]=z;
}
int main()
{
    cin>>n>>m>>s;
    for (int i=1;i<=n;i++) fi[i]=-1;
    for (int i=1;i<=m;i++)
    {
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        edge(u,v,w);
    }
    for (int i=1;i<=n;i++) dis[i]=inf;
    dis[s]=0;
    memset(f,0,sizeof(f));  
    int temp=s;
    for (int turn=1;turn<=n-1;turn++)
    {
        f[temp]=1;
        for (int i=fi[temp];i!=-1;i=ne[i]) if ((!f[v[i]])&&(dis[temp]+w[i]<dis[v[i]]))
        {
            dis[v[i]]=dis[temp]+w[i];
        }
        int mi=inf;
        for (int i=1;i<=n;i++) if ((!f[i])&&(dis[i]<mi))
        {
            mi=dis[i];
            temp=i;
        }       
    }   
    for (int i=1;i<=n;i++) printf("%d ",dis[i]);
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/reshuffle/p/12188979.html
今日推荐