A - 图结构练习——最短路径

朴素版dijkstra

#include<bits/stdc++.h>
#define endl '\n'
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;
const int N=110;
int g[N][N];
bool vis[N];
int dist[N];
int n,m;
void init()
{
    for(int i=0;i<=n;i++)
    {
        dist[i]=INF;
        vis[i]=false;
        for(int j=0;j<=n;j++)
        {
            if(i==j)g[i][j]=0;
            else g[i][j]=INF;
        }
    }
}
void dijkstra()
{
    dist[1]=0;
    for(int i=0;i<n;i++)
    {
        int t=-1;
        for(int j=1;j<=n;j++)
        {
            if(!vis[j]&&(t==-1||dist[j]<dist[t]))
                t=j;
        }
        vis[t]=true;
        for(int j=1;j<=n;j++)
        {
            if(!vis[j])
            {
                dist[j]=min(dist[j],dist[t]+g[t][j]);
            }
        }
    }
}
int main()
{
    while(cin>>n>>m)
    {
        init();
        for(int i=0;i<m;i++)
        {
            int u,v,w;
            cin>>u>>v>>w;
            g[u][v]=min(g[u][v],w);
            g[v][u]=min(g[v][u],w);
        }
        dijkstra();
        cout<<dist[n]<<endl;
    }
    return 0;
}

堆优化dijkstra

#include<bits/stdc++.h>
#define endl '\n'
#define INF 0x3f3f3f3f
using namespace std;
const int N=110;
typedef pair<int,int> PII;
vector<PII> g[N];
int dist[N];
bool vis[N];
int n,m;
void init()
{
    for(int i=0;i<=n;i++)
    {
        g[i].clear();
        dist[i]=INF;
        vis[i]=false;
    }
}
void dijkstra()
{
    priority_queue<PII,vector<PII>,greater<PII>> q;
    dist[1]=0;
    q.push({dist[1],1});
    while(q.size())
    {
        int u=q.top().second;
        q.pop();
        if(vis[u])continue;
        vis[u]=true;
        for(auto [v,w]:g[u])
        {
            if(dist[v]>dist[u]+w)
            {
                dist[v]=dist[u]+w;
                q.push({dist[v],v});
            }
        }
    }
}
int main()
{
    while(cin>>n>>m)
    {
        init();
        for(int i=0;i<m;i++)
        {
            int u,v,w;
            cin>>u>>v>>w;
            g[u].push_back({v,w});
            g[v].push_back({u,w});
        }
        dijkstra();
        cout<<dist[n]<<endl;
    }
    return 0;
}

一定不要忘了初始化dist，g，vis