NEFUOJ 208 宫锁珠帘（最短路径问题）

http://acm.nefu.edu.cn/JudgeOnline/contestshow.php?contest_id=120&problem_id=208&num=3

description

新年期间湖南卫视又上映了不少宫闱大戏，晴川穿走了，又来了个宫锁珠帘的袁珊珊，袁珊珊不够美，没锁住观众的心，特别是咱们整ACM的，对这些宫闱大戏当然是不屑一顾了，可这新亮剑的李云龙咱们还是有得一看的，最近听说A地又驻扎了一个骑兵营，李云龙心里就乐了，心想老子吃了骑兵营这么多亏，这下终于有机会自个儿也能整个啥骑兵营威风威风了，于是拿出地图决定抄最短的路立马拿下这个骑兵营。

input

每组数据第一行包含两个正整数N和M(0 < N < 100,0 < M < 100)，分别代表李云龙现有地图上显示的村子或者驻扎地的总数目，分别以0～N-1编号。 接下来是M行道路信息。每一行有三个整数A,B,X(0 <= A,B < N,A!=B,0< X < 10000),表示A和B之间有一条长度为X的双向道路。 再接下一行有两个整数S,T(0 <= S,T < N)，分别代表起点和终点。

output

输出从S到T的最短路的长度。若不存在这样的一条路，则输出-1.

sample_input

5 4 0 1 5 0 3 4 3 2 7 3 2 6 0 2 4 1 2 3 1 1 2

sample_output

10 -1

和上一题就差一点。。。。

这是我的代码：

#include <stdio.h>
#include <iostream>
#include <string.h>
#define MAX 99999
#define LEN 101
using namespace std;
int map[LEN][LEN];
void abc()
{
    int i,j;
    for(i=0;i<LEN ;i++)
    {
        for(j=0;j<LEN;j++)
          map[i][j]=MAX;
    }
}
void prime(int n)
{
    int i,j,k;
    for(k=0;k<n;k++)
       for(i=0;i<n;i++)
          for(j=0;j<n;j++)
         if(map[i][j]>map[i][k]+map[k][j])
            map[i][j]=map[i][k]+map[k][j];
}
int main()
{
    abc();
    int m,n,a,b,c,d,w,i;
    while(~scanf("%d%d",&n,&m))
    {
        for(i=0;i<m;i++)
        {
            cin>>a>>b>>w;
            if(map[a][b]>w)
              map[a][b]=w;
        }
        cin >> c>> d;
        prime(n);
        if(map[c][d]==MAX)
           cout << "-1\n";
        else
        cout << map[c][d] << endl;
    }
    return 0;
}