J - 热浪

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

#include
#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>>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>u>>v>>w;
        g[u][v]=min(g[u][v],w);//取两点之间的最小距离,来处理重边和自环
        g[v][u]=min(g[v][u],w);//双向边
    }
    dijkstra();
    cout<

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

#include
using namespace std;
const int N=2510;
int n,m,s,t;
typedef pair PII;//first为邻点,second为两点之间的边权
vector 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,greater>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>u>>v>>w;
        g[u].push_back({v,w});
        g[v].push_back({u,w});
    }
    dijkstra();
    cout<

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

其他算法补充:

#include 
#define endl '\n'
#define INF 0x3f3f3f3f
using namespace std;
const int N=2510;
typedef pair pii;
vector 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,greater> 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;iq;
    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>u>>v>>w;
        g[u].push_back({v,w});
        g[v].push_back({u,w});
    }
    init();
    spfa();
    //dijkstra();
    //bellman();
    cout<

你可能感兴趣的:(2022寒假第一次结训赛,算法)