保研机试备考十：最小生成树、最短路

保研机试备考十：最小生成树、最短路

AcWing 858. Prim算法求最小生成树

题目

https://www.acwing.com/problem/content/860/

思路

参考我初学时的博客https://blog.csdn.net/weixin_45798993/article/details/122813952

代码

#include<iostream>
#include<cstring>
using namespace std;
const int N=510,INF=0x3f3f3f3f;
bool st[N];
int g[N][N];
int dist[N];
int n,m;
int prim()
{
    memset(dist,0x3f,sizeof(dist));
    
    int res=0;
    
    for(int i=0;i<n;i++)
    {
        int t=-1;
        for(int j=1;j<=n;j++)
            if(!st[j]&&(t==-1||dist[t]>dist[j]))
                t=j;
        if(i&&dist[t]==INF) return INF;
        if(i) res+=dist[t];
        st[t]=true;
        for(int j=1;j<=n;j++)
            if(!st[j])
                dist[j]=min(dist[j],g[t][j]);
    }
    return res;
}
int main()
{
    memset(g,0x3f,sizeof(g));
    scanf("%d%d",&n,&m);
    int u,v,w;
    for(int i=0;i<m;i++)
    {
        scanf("%d%d%d",&u,&v,&w);
        g[u][v]=g[v][u]=min(g[u][v],w);
    }
    int t=prim();
    if(t==INF) printf("impossible");
    else printf("%d",t);
    return 0;
}

---

AcWing 849. Dijkstra求最短路 I

题目

https://www.acwing.com/problem/content/851/

思路

参考我初学时的博客https://blog.csdn.net/weixin_45798993/article/details/122791024

代码

#include<iostream>
#include<cstring>
using namespace std;
const int N=510,INF=0x3f3f3f3f;
bool st[N];
int g[N][N];
int dist[N];
int n,m;
int Dijkstra()
{
    memset(dist,0x3f,sizeof(dist));
    st[1]=true;
    dist[1]=0;
    for(int i=2;i<=n;i++) dist[i]=g[1][i];
    for(int i=0;i<n;i++)
    {
        int t=-1;
        for(int j=1;j<=n;j++)
            if(!st[j]&&(t==-1||dist[t]>dist[j]))
                t=j;
        st[t]=true;
        if(t==n) break;   //如果已经找到了1到n的距离，就赶尽break出来
        for(int j=1;j<=n;j++)
            if(!st[j])
                dist[j]=min(dist[j],dist[t]+g[t][j]);
    }
    if(dist[n]==INF) return -1;
    else return dist[n];
}
int main()
{
    memset(g,0x3f,sizeof(g));
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) g[i][i]=0;
    int x,y,z;
    for(int i=0;i<m;i++)
    {
        scanf("%d%d%d",&x,&y,&z);
        if(x!=y)
            g[x][y]=min(g[x][y],z);
    }
    printf("%d",Dijkstra());
    return 0;
}

---

AcWing 854. Floyd求最短路

题目

https://www.acwing.com/problem/content/856/

思路

参考初学时的博客https://blog.csdn.net/weixin_45798993/article/details/122791024

代码

#include<iostream>
#include<cstring>
using namespace std;
const int N=210,INF=0x3f3f3f3f;
int n,m,q;
int d[N][N];
void floyd()
{
    for(int k=1;k<=n;k++)
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                d[i][j]=min(d[i][j],d[i][k]+d[k][j]);
}
int main()
{
    scanf("%d%d%d",&n,&m,&q);
    memset(d,0x3f,sizeof(d));
    for(int i=1;i<=n;i++) d[i][i]=0;
    int x,y,z;
    for(int i=0;i<m;i++)
    {
        scanf("%d%d%d",&x,&y,&z);
        d[x][y]=min(d[x][y],z);
    }
    floyd();
    while(q--)
    {
        scanf("%d%d",&x,&y);
        if(d[x][y]>INF/2) puts("impossible");
        else printf("%d\n",d[x][y]);
    }
    return 0;
}

---