HDOJ-----1596最短路

find the safest road

Time Limit: 10000/5000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 11623    Accepted Submission(s): 4106

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
  
  

这个要把模板做一点小改动，把dis（权值）的初值改一下，从无穷大改为0，还有输入，val的值改成了double型的，出发点即第一个遍历点的dis初值也修改成了1（没有出发点等于目的地的情况，就不多加考虑了），不过复杂度很高，不知道别人怎么优化到800+ms的，我只能优化到1800+ms了

#include<cstdio>
#include<queue>
#include<algorithm>
#include<cstring>
#define maxn 1000100
#define inf 1e-9
using namespace std;
int head[maxn];
bool vis[maxn];
double dis[maxn];
int m, n, ans, ok, num;
struct node{
	int from, to, next;
	double val;
}edge[maxn*2];
void add(int u, int v, double w){
	node e = {u, v, head[u], w};
	edge[num] = e;
	head[u] = num++;
}
void dij(int t){
	memset(dis, 0, sizeof(dis));
	memset(vis, false, sizeof(vis));
	queue <int > Q;
	dis[t] = 1;
	Q.push(t);
	while(!Q.empty()){
		int u = Q.front();
		Q.pop();
		vis[u] = false;
		for(int i = head[u]; i != -1; i = edge[i].next){
			int v = edge[i].to;
			if(dis[v] < dis[u] * edge[i].val){//还有这里的判断改为小于号，+变为*，跟最长路有点像了
				dis[v] = dis[u] * edge[i].val;
				if(!vis[v]){
					vis[v] = true;
					Q.push(v);
				}
			}
		}
	}
}
int main(){
	int c, a, b, x, y, T;
	double k;
	while(~scanf("%d", &c)){
		num = 0;
		memset(head, -1, sizeof(head));
		for(int i = 1; i <= c; i++){
			for(int j = 1; j <= c; j++){
				scanf("%lf", &k);
				if(j <= i || k == 0){
					continue;
				}
				add(i, j, k);
				add(j, i, k);
			}
		}
		scanf("%d", &T);
		while(T--){
			scanf("%d%d", &x, &y);
			dij(x);
			if(dis[y] == 0){
				printf("What a pity!\n");
				continue;
			}
			printf("%.3lf\n", dis[y]);
		}
	}
}