C--最短路 SDUT OJ

C–最短路
Time Limit: 7000 ms Memory Limit: 65536 KiB
Submit Statistic
Problem Description

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

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

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

3 1
1 2 3
1 2
Sample Output

3

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