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\n",ans);
            else puts("What a pity!");
        }
    }
}