hihocoder：最短路径

1081 : 最短路径·一

时间限制: 10000ms

单点时限: 1000ms

内存限制: 256MB

描述

万圣节的早上，小Hi和小Ho在经历了一个小时的争论后，终于决定了如何度过这样有意义的一天——他们决定去闯鬼屋！

在鬼屋门口排上了若干小时的队伍之后，刚刚进入鬼屋的小Hi和小Ho都颇饥饿，于是他们决定利用进门前领到的地图，找到一条通往终点的最短路径。

鬼屋中一共有N个地点，分别编号为1..N，这N个地点之间互相有一些道路连通，两个地点之间可能有多条道路连通，但是并不存在一条两端都是同一个地点的道路。那么小Hi和小Ho至少要走多少路程才能够走出鬼屋去吃东西呢？

提示：顺序！顺序才是关键。

输入

每个测试点（输入文件）有且仅有一组测试数据。

在一组测试数据中：

第1行为4个整数N、M、S、T，分别表示鬼屋中地点的个数和道路的条数，入口（也是一个地点）的编号，出口（同样也是一个地点）的编号。

接下来的M行，每行描述一条道路：其中的第i行为三个整数u_i, v_i, length_i，表明在编号为u_i的地点和编号为v_i的地点之间有一条长度为length_i的道路。

对于100%的数据，满足N<=10^3，M<=10^4, 1 <= length_i <= 10^3, 1 <= S, T <= N, 且S不等于T。

对于100%的数据，满足小Hi和小Ho总是有办法从入口通过地图上标注出来的道路到达出口。

输出

对于每组测试数据，输出一个整数Ans，表示那么小Hi和小Ho为了走出鬼屋至少要走的路程。

样例输入

5 23 5 4
1 2 708
2 3 112
3 4 721
4 5 339
5 4 960
1 5 849
2 5 98
1 4 99
2 4 25
2 1 200
3 1 146
3 2 106
1 4 860
4 1 795
5 4 479
5 4 280
3 4 341
1 4 622
4 2 362
2 3 415
4 1 904
2 1 716
2 5 575

样例输出

123

import java.util.Arrays;
import java.util.HashSet;
import java.util.Scanner;

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
/**
 *
 * @author Administrator
 */
public class Main {
    static int[][] map;
    static Scanner cin ;
    static int n, m, s, t;
    static HashSet<Integer> set;
    
    public static void main(String[] args) {
        init();
        int dp[] = new int[n + 1];
//        boolean[] vis = new boolean[n + 1];
        Arrays.fill(dp, Integer.MAX_VALUE);
        for(int i = 1; i <= n; ++i){
            dp[i] = Math.min(dp[i], getDistance(s, i));
//            System.out.println(i + "   " + dp[i]);
        }
        set.remove(s);
        for(int i = 1; i < n; ++i){
            int idx = getMinIndex(dp);
//            System.out.println("idx = " + idx);
            if(idx == t){
                break;
            }
            set.remove(idx);
            for(int k : set){
                int dis = getDistance(idx, k);
                if(dis == Integer.MAX_VALUE){
                    continue;
                }
                dp[k] = Math.min(dis + dp[idx], dp[k]);
            }
        }
        
        System.out.println(dp[t]);
    }
    
    public static int getDistance(int i, int j){
        return Math.min(map[i][j], map[j][i]);
    }
    
    public static int getMinIndex(int[] a){
        int minIndex = -1;
        int curMinValue = Integer.MAX_VALUE;
        for(int i : set){
            if(a[i] < curMinValue){
                curMinValue = a[i];
                minIndex = i;
            }
        }
        return minIndex;
    }
    
    public static void init(){
        cin = new Scanner(System.in);
        n = cin.nextInt();
        m = cin.nextInt();
        s = cin.nextInt();
        t = cin.nextInt();
        map = new int[n + 1][n + 1];
        set = new HashSet<>();
        for(int i = 1; i < map.length; ++i){
            Arrays.fill(map[i], Integer.MAX_VALUE);
            set.add(i);
        }
//        System.out.println(set);
        for(int i = 0; i < m; ++i){
            int t1 = cin.nextInt();
            int t2 = cin.nextInt();
            int len = cin.nextInt();
            map[t1][t2] = Math.min(map[t1][t2], len);
        }
    }
}