C--最短路 (spfa算法)

来源:互联网 发布:37传奇霸业 血符数据 编辑:程序博客网 时间:2024/05/29 18:20

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
3

spfa算法详解(http://blog.csdn.net/muxidreamtohit/article/details/7894298)

#include<bits/stdc++.h>#define Maxn 0x3f3f3f3fusing namespace std;int n, m, cnt;int vis[500010], dis[500010],head[500010];struct node{    int v, w;    int next;}edge[4000010];void add(int u, int v, int w){    edge[cnt].v = v;    edge[cnt].w = w;    edge[cnt].next = head[u];    head[u] = cnt++;}void spfa(int s, int e){    queue<int>q;    memset(vis,0,sizeof(vis));    for(int i = 1; i <= n; i++)        dis[i] = Maxn;    dis[s] = 0;    vis[s] = 1;    q.push(s);    while(!q.empty())    {        int k = q.front();        q.pop();        vis[k] = 0;        for(int i = head[k]; i != -1; i = edge[i].next)        {            int v = edge[i].v;            if(dis[v] > dis[k] + edge[i].w)            {                dis[v] = dis[k] + edge[i].w;                if(!vis[v])                {                    vis[v] = 1;                    q.push(v);                }            }        }    }}int main(){    while(cin >> n >> m)    {        memset(head,-1,sizeof(head));        cnt = 0;        while(m--)        {            int u, v, w;            cin >> u >> v >> w;            add(u,v,w);            add(v,u,w);        }        int s, e;        cin >> s >> e;        spfa(s,e);        cout << dis[e] << endl;    }    return 0;}