J - 热浪

以朴素版dijkstra(稠密图)(邻接矩阵)解

#include<bits/stdc++.h>
#define endl '\n'
using namespace std;
const int N=2510;
const int INF=0x3f3f3f3f;//以0x3f3f3f3f代表正无穷
int n,m,S,T,u,v,w;
int g[N][N];
int dist[N];
bool st[N];
void dijkstra()
{
    dist[S]=0;
    for(int i=0;i<n;i++)
    {
        int t=-1;
        for(int j=1;j<=n;j++)
        {
            if(!st[j]&&(t==-1||dist[j]<dist[t]))//寻找一个还未确定最短距离的点
                t=j;
        }
        st[t]=true;//更新此点的状态为已确定最短距离
        for(int j=1;j<=n;j++)//用此点来更新其他点
        {
            dist[j]=min(dist[j],dist[t]+g[t][j]);//比较通过此点到达目标点的最短距离或者目标点原来的最短距离,取最小值
        }
    }
}
int main()
{
    cin>>n>>m>>S>>T;
    for(int i=0;i<=n;i++)
    {
        dist[i]=INF;
        for(int j=0;j<=n;j++)
        {
            if(i==j)g[i][j]=0;//初始化自己到自己的距离为0
            else g[i][j]=INF;//其余为正无穷
        }
    }
    for(int i=0;i<m;i++)
    {
        cin>>u>>v>>w;
        g[u][v]=min(g[u][v],w);//取两点之间的最小距离,来处理重边和自环
        g[v][u]=min(g[v][u],w);//双向边
    }
    dijkstra();
    cout<<dist[T];
    return 0;
}

堆优化版dijkstra(稀疏图)(邻接表)解

#include<bits/stdc++.h>
using namespace std;
const int N=2510;
int n,m,s,t;
typedef pair<int,int> PII;//first为邻点,second为两点之间的边权
vector<PII> g[N];//邻接表存储图
int dist[N];//记录每个点的最小距离
bool vis[N];//判断此点状态,是否已确定最小距离
void dijkstra()
{
    for(int i=0;i<=n+1;i++)
    {
        vis[i]=false;
        dist[i]=1e9;
    }
    dist[s]=0;
    priority_queue<PII,vector<PII>,greater<PII>>q;//q存储的是个pair
    //first为此点到起点的最短距离,second为此点的编号
    //用堆来维护还未确定最小距离的点中距离起点距离最小的点
    //由于小根堆会优先按照第一个数据进行排序,所以要把距离放到first的位置
    q.push({dist[s],s});
    while(q.size())
    {
        int u=q.top().second;//取出此编号
        q.pop();
        if(vis[u])continue;//如果此编号已经被确定最小距离则为冗余备份,跳过此次更新
        vis[u]=true;//更新此点状态为已确定最小距离
        for(auto [v,w]:g[u])//用u的最小距离来更新u的所有临边连接的点的最小距离
        {
            if(dist[v]>dist[u]+w)
            {
                dist[v]=dist[u]+w;
                q.push({dist[v],v});
            }
        }
    }
}
int main()
{
    cin>>n>>m>>s>>t;
    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[t];
    return 0;
}

由于此题边数不超过1e4所以更宜用堆优化版解答(点也不多,朴素版也能过)

其他算法补充:

#include <bits/stdc++.h>
#define endl '\n'
#define INF 0x3f3f3f3f
using namespace std;
const int N=2510;
typedef pair<int,int> pii;
vector<pii> g[N];
int dist[N];
bool vis[N];
int n,m,s,t;
void init()
{
    for(int i=0;i<=n;i++)
    {
        vis[i]=false;
        dist[i]=INF;
    }
}
void dijkstra()
{
    dist[s]=0;
    priority_queue<pii,vector<pii>,greater<pii>> q;
    q.push({dist[s],s});
    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});
            }
        }
    }
}
void bellman()//bellman不用vis数组判断点
{
    dist[s]=0;
    for(int i=0;i<n;i++)
    {
        for(int u=1;u<=n;u++)
        {
            for(auto [v,w]:g[u])
            {
                dist[v]=min(dist[v],dist[u]+w);
            }
        }
    }
}
void spfa()
{
    dist[s]=0;
    vis[s]=true;//放入队列更新状态为待松弛
    queue<int>q;
    q.push(s);
    while(q.size())
    {
        int u=q.front();
        q.pop();
        vis[u]=false;//已松弛当前结点
        for(auto [v,w]:g[u])
        {
            if(dist[v]>dist[u]+w)
            {
                dist[v]=dist[u]+w;
                if(!vis[v])//如果更新结点未在队列中,则将该点加入队列
                {
                    q.push(v);
                    vis[v]=true;
                }
            }
        }
    }
}
int main()
{
    cin>>n>>m>>s>>t;
    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});
    }
    init();
    spfa();
    //dijkstra();
    //bellman();
    cout<<dist[t];
    return 0;
}

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值