HDOJ find the safest road 1596【最短路变形】

find the safest road Time Limit: 10000/5000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others) Total Submission(s): 9033    Accepted Submission(s): 3173 Problem Description 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   Author ailyanlu   Source HDU 2007-Spring Programming Contest - Warm Up （1）   Recommend 8600   |   We have carefully selected several similar problems for you:   1217  1598  1142  1690  1385

最短路变形，把求最小改成最大即可。

注意在Q次询问中会发生权值改变，所以我们需要定义一个数组来保存权值，询问之前把值重新赋给权值

#include <stdio.h>
#include <math.h>
#include <vector>
#include <queue>
#include <string>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
#define INF 0x3f3f3f3f

using namespace std;

const int MAXN = 1010;
bool used[MAXN];
double d[MAXN];
double cost[MAXN][MAXN];
double a[MAXN][MAXN];
int N;

void Dijkstra(int s)
{
    for(int i=0;i<N;i++){
        used[i]=false;
        d[i]=0;
    }
    d[s]=1;
    while(true){
        int v=-1;
        for(int i=0;i<N;i++)
            if(!used[i]&&(v==-1||d[v]<d[i])) v=i;
        if(v==-1) break;
        used[v]=true;
        for(int i=0;i<N;i++)
            d[i]=max(d[i],d[v]*cost[v][i]);
    }
}
int main()
{
    while(scanf("%d",&N)!=EOF){
        for(int i=0;i<N;i++){
            for(int j=0;j<N;j++){
                scanf("%lf",&a[i][j]);
                //if(a[i][j]==0) a[i][j]=0;
            }
        }
        int Q;
        scanf("%d",&Q);
        int s,g;
        while(Q--){
            //memcpy(cost,a,sizeof(a));//可以代替下面的for循环，但是耗时
            for(int i=0;i<N;i++){
                for(int j=0;j<N;j++)
                    cost[i][j]=a[i][j];
            }
            scanf("%d%d",&s,&g);
            Dijkstra(s-1);
            if(d[g-1]==0) printf("What a pity!\n");
            else printf("%.3lf\n",d[g-1]);
        }
    }
    return 0;
}