HDU 1874 畅通工程续 （Dijkstra算法）

 1 #include <iostream>

 2 

 3 using namespace std;

 4 

 5 #define MAX_VERTEXT 250

 6 const int MAX_WEIGHT= 0x7f7f7f7f;

 7 

 8 int map[MAX_VERTEXT][MAX_VERTEXT];

 9 int path[MAX_VERTEXT];

10 

11 void Init()

12 {

13     memset(map,MAX_WEIGHT,sizeof(map));

14     memset(path,MAX_WEIGHT,sizeof(path));

15 }

16 

17 void Input(int m)

18 {

19     int start,end,weight;

20     for (int i=0;i<m;i++)

21     {

22         cin>>start>>end>>weight;                    

23         //这里太BT了。。。

24         if (weight<map[start][end])

25             map[start][end]=map[end][start]

26                             =weight;

27         if(start==end)

28             map[start][end]=0;

29     }

30 }

31 

32 template<typename T>

33 T min(const T& x,const T& y)

34 {

35     return x<y ? x:y;

36 }

37 

38 int Dijkstra(int n,int s,int e)

39 {

40     int i,j,k;

41     int v[MAX_VERTEXT]={0};

42     for (path[s]=0,i=0;i<n;i++)

43     {

44         //把顶点分成两个集合S，U（U为剩余顶点集）

45         //一开始S中只有s顶点（开始点）

46         //从剩余的顶点中挑选路径最短的加入S

47         for(k=-1,j=0;j<n;j++)

48             if(!v[j]&&(k==-1||path[j]<path[k]))

49                 k=j;

50 

51         for(v[k]=1,j=0;j<n;j++)

52             if(!v[j]&&(map[k][j]!=MAX_WEIGHT))

53                 path[j]=::min(path[k]+map[k][j],path[j]);

54     }

55     if(path[e]>=MAX_WEIGHT)

56         return -1;

57     else

58         return path[e];

59 }

60 

61 int main()

62 {

63     int N,M,start,end;

64     while(cin>>N>>M)

65     {

66         Init();

67         Input(M);

68         cin>>start>>end;

69         if(start==end)

70             cout<<"0"<<endl;

71         else

72             cout<<Dijkstra(N,start,end)<<endl;

73     }

74     return 0;

75 }