HDU 1874 畅通工程续

畅通工程续

Time Limit: 3000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 40454 Accepted Submission(s): 14964

Problem Description
某省自从实行了很多年的畅通工程计划后，终于修建了很多路。不过路多了也不好，每次要从一个城镇到另一个城镇时，都有许多种道路方案可以选择，而某些方案要比另一些方案行走的距离要短很多。这让行人很困扰。

现在，已知起点和终点，请你计算出要从起点到终点，最短需要行走多少距离。

Input
本题目包含多组数据，请处理到文件结束。
每组数据第一行包含两个正整数N和M(0

#include <stdio.h>
#include <algorithm>
#include <string.h>
#include <iostream>
#define N 201
#define INF 19999
#include <iostream>
using namespace std;
int dis[N],vis[N],way[N][N];
void init(int n)
{
    for(int i=0; i<n; i++)
    {
        for(int j=0;j<n;j++)
        {
            if(i==j)
                way[i][j]=0;
            else way[i][j]=INF;
        }
        vis[i]=0;
        dis[i]=INF;
    }
}
void dijkstr(int n,int fir,int ends)
{
    for(int i=0; i<n; i++)
    {
        dis[i]=way[fir][i];
    }

    int min_post,min_value;
    dis[fir]=0;
    for(int i=0; i<n; i++)
    {
        min_value=INF;
        for(int j=0; j<n; j++)
        {
            if(dis[j]<min_value&&(!vis[j]))
            {
                min_post=j;
                min_value=dis[j];
            }
        }
        vis[min_post]=1;
        for(int k=0; k<n; k++)
        {
            if(!vis[k])
            dis[k]=min(dis[k],dis[min_post]+way[min_post][k]);
        }
    }
}

int main()
{
    int n,m;
    while(~scanf("%d %d",&n,&m))
    {
        init(n);
        //cout<<"dis[n-1] is "<<dis[n-1]<<endl;
        int str,ends,len;
        init(n);
        for(int i=0; i<m; i++)
        {
            scanf("%d %d %d",&str,&ends,&len);
            if(way[str][ends]>len)
                way[str][ends]=way[ends][str]=len;
        }
        //cout<<"The second dis[n-1] is "<<dis[n-1]<<endl;
        scanf("%d %d",&str,&ends);
        dijkstr(n,str,ends);
        if(str==ends)
         {
             printf("0\n");
             continue;
          }
        else if(dis[ends]!=INF)
            cout<<dis[ends]<<endl;
        else cout<<"-1"<<endl;
    }
    return 0;
}