HDU-1596 find the safest road

XX星球有很多城市，每个城市之间有一条或多条飞行通道，但是并不是所有的路都是很安全的，每一条路有一个安全系数s,s是在 0 和 1 间的实数(包括0，1)，一条从u 到 v 的通道P 的安全度为Safe(P) = s(e1)*s(e2)…*s(ek) e1,e2,ek是P 上的边 ，现在8600 想出去旅游，面对这这么多的路，他想找一条最安全的路。但是8600 的数学不好，想请你帮忙 ^_^
Input
输入包括多个测试实例，每个实例包括：
第一行：n。n表示城市的个数n<=1000;
接着是一个n*n的矩阵表示两个城市之间的安全系数，(0可以理解为那两个城市之间没有直接的通道)
接着是Q个8600要旅游的路线,每行有两个数字，表示8600所在的城市和要去的城市
Output
如果86无法达到他的目的地，输出”What a pity!”,
其他的输出这两个城市之间的最安全道路的安全系数,保留三位小数。
Sample Input
3
1 0.5 0.5
0.5 1 0.4
0.5 0.4 1
3
1 2
2 3
1 3
Sample Output
0.500
0.400
0.500
可以理解为求最长路，迪杰斯特拉算法倒过来用就可以了，注意的是权值不再是累加，而是累乘，每次选择安全系数最大的点就可以了；

#include<bits/stdc++.h>

using namespace std;
const int N = 1000 + 5;
double mat[N][N],D[N];
bool visit[N];

void Init_mat(){
    for(int i=0;i<N;i++){
        for(int j=0;j<N;j++) mat[i][j] = 0;
    }
}

double solve_question(int n,int s,int e){
    int cnt,i,k;
    for(i=0;i<N;i++) D[i] = 0,visit[i] = false;
    D[s] = 1;
    for(cnt=1;cnt<=n;cnt++){
        for(k=-1,i=1;i<=n;i++)
            if(!visit[i]&&(k==-1 || D[i]>D[k])) k =i;
        for(visit[k] = true,i=1;i<=n;i++)
            if(!visit[i] && D[i] < D[k]*mat[k][i])
            D[i] = D[k]*mat[k][i];
    }
    return D[e];
}

int main(){
    int n,m;
    while(scanf("%d",&n)==1){
        Init_mat();
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++) scanf("%lf",&mat[i][j]);
        int u,v;
        scanf("%d",&m);
        for(int i=0;i<m;i++){
            scanf("%d %d",&u,&v);
            double ans = solve_question(n,u,v);
            if(ans!=0)
            printf("%.3lf
",ans);
            else puts("What a pity!");
        }
    }
}