C--最短路(Bellman-Ford算法)

最新推荐文章于 2024-02-21 10:09:27 发布

sduthao

最新推荐文章于 2024-02-21 10:09:27 发布

阅读量569

点赞数

分类专栏：最短路

本文链接：https://blog.csdn.net/sduthao/article/details/76220235

版权

最短路专栏收录该内容

5 篇文章 0 订阅

订阅专栏

C--最短路

Time Limit: 7000MS Memory Limit: 65536KB

Problem Description

给出一个带权无向图，包含n个点，m条边。求出s，e的最短路。保证最短路存在。

Input

多组输入。

对于每组数据。

第一行输入n，m(1<= n && n<=5*10^5,1 <= m && m <= 2*10^6)。
接下来m行，每行三个整数，u,v,w,表示u，v之间有一条权值为w(w >= 0)的边。
最后输入s，e。

Output

对于每组数据输出一个整数代表答案。

Example Input

3 1
1 2 3
1 2

Example Output

Hint

Author

zmx

#include <bits/stdc++.h>

using namespace std;

int dis[500004], i, k, n, m, check, u[4000004], v[4000004], w[4000004];
void Bellman();
int main()
{
    int s, e, x, y, z, t;
    while(scanf("%d %d", &n, &m) != EOF)
    {
        t = 1;
        for(i = 1; i <= m; i++)
            {
            scanf("%d %d %d", &x, &y, &z);
            u[t] = x, v[t] = y, w[t] = z;
            t++;
            u[t] = y, v[t] = x, w[t] = z;
            t++;
            }
        scanf("%d %d", &s, &e);
        m = m * 2;
        for(i = 1; i <= n; i++)
            dis[i] = INT_MAX;
        dis[s] = 0;
        Bellman();
        printf("%d\n", dis[e]);
    }
    return 0;
}
void Bellman()   //Bellman-Ford 算法
{
    for(k = 1; k < n; k++)
        {
            check = 0;
            for(i = 1; i <= m; i++)
                {
                    if(dis[u[i]] != INT_MAX && dis[v[i]] > dis[u[i]] + w[i]) //如果不判断dis是否为最大值，可能会溢出
                        {
                            dis[v[i]] = dis[u[i]] + w[i];
                            check = 1;
                        }
                }
            if(check == 0)
                break;
        }
        return ;
}

队列优化后

#include <bits/stdc++.h>
#define inf 0x3f3f3f3f
using namespace std;

int u[4000005], v[4000005], w[4000005];
int first[500005], next[4000005];
int dis[500005], book[500005];

int main()
{
    int x, y, z, n, m, s, e, t, i, k;
    while(scanf("%d %d", &n, &m) != EOF)
    {
        t = 1;
        queue<int>q;
        for(i = 1; i <= n; i++)
            {
            dis[i] = inf;
            first[i] = -1;
            book[i] = 0;
            }
        for(i = 1; i <= m; i++)
            {
            scanf("%d %d %d", &x, &y, &z);
            u[t] = x, v[t] = y, w[t] = z;
            next[t] = first[u[t]];
            first[u[t]] = t;
            t++;
            u[t] = y, v[t] = x, w[t] = z;
            next[t] = first[u[t]];
            first[u[t]] = t;
            t++;
            }
        m = m * 2;
        scanf("%d %d", &s, &e);
        dis[s] = 0;
        q.push(s);
        book[s] = 1;
        while(q.size() > 0)
        {
            k = first[q.front()];
            book[q.front()] = 0;
            q.pop();
            while(k != -1)
            {
                if(dis[v[k]] > dis[u[k]] + w[k])
                    {
                        dis[v[k]] = dis[u[k]] + w[k];
                        if(book[v[k]] == 0)
                            {
                                q.push(v[k]);
                                book[v[k]] = 1;
                            }
                    }
                k = next[k];
            }
        }
        printf("%d\n", dis[e]);
    }
    return 0;
}