HDU 1874 畅通工程续

http://acm.hdu.edu.cn/showproblem.php?pid=1874

第一次知道有个东西叫重边。。。囧

#include <iostream>
#include <string.h>
using namespace std;
const int INF=1000000000;
int map[210][210];
int n,m,f;
int dis[210],vis[210];
int Dijkstra(int s,int t)
{
    for(int i=0;i<n;i++)
        dis[i]=INF;
    memset(vis,0,sizeof(vis));
    dis[s]=0;
    for(int i=0;i<n;i++)
    {
        int ans=INF,u;
        for(int j=0;j<n;j++)
            if(!vis[j] && dis[j]<ans)
            {
                ans=dis[j];
                u=j;
            }
        vis[u]=1;
        if(u==t){
            f=1;
            return ans;    
        } 
        for(int j=0;j<n;j++)
            if(map[u][j]!=INF && dis[u]+map[u][j]<dis[j])
                dis[j]=dis[u]+map[u][j];
    }
}
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        f=0;
        int s,t,k;
        for(int i=0;i<=n;i++)
            for(int j=0;j<=n;j++)
                map[i][j]=INF;
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&s,&t,&k);
            if(map[s][t]>k)
                map[s][t]=map[t][s]=k;
        }
        int a,b;
        scanf("%d%d",&a,&b);
        int ans=Dijkstra(a,b);
        if(f)
            printf("%d\n",ans);
        else
            puts("-1");
    }
    return 0;
}