POJ 3268 Silver Cow Party dijkstra最短路

POJ 3268 Silver Cow Party

题目大意：

每个农场有一只奶牛，所有奶牛都要去指定的一个农场X中聚会，聚会结束后各自回自己的农场，
求所有的奶牛出门参加聚会再回到家中的最短用时，即求出耗时最长的奶牛的时间

具体思路：

最短路求解，先求x到各个农场的时间，再将各边反向，**再求一次x到各个农场的时间。**将两个时间相加，最小的时间即为所求
我本来还傻傻的跑了N遍，以求出每个点对其他点的最短路，看了题解才发现有这样神奇的思路。
求图中所有点到指定点的最短距离可以转化为求指定点到所有点的最短距离，有向图要反向

AC代码：

#include <iostream>
#include <string.h>
#include <algorithm>
#define inf 0x3f3f3f3f
using namespace std;
const int N = 1010;
int dis[N][N], n, m, k, dis1[N][N];
bool vis[N];
struct node {
	int d[N];
}cow[N];
int sum[N];

// st 从1到n遍历 
// 传入不同的dis数组
void dijkstra(int distance[][N], int st) {
	memset(cow[st].d, inf, sizeof(cow[st].d));
	memset(vis, 0, sizeof vis);
	cow[st].d[st] = 0;
	for(int j = 1; j < n; j++) {
		int minid, mindis = inf;
		for(int i = 1; i <= n; i++) {
			if(!vis[i] && cow[st].d[i] < mindis) {
				mindis = cow[st].d[i];
				minid = i;
			}
		}
		vis[minid] = true;
		for(int i = 1; i <= n; i++) {
			if(cow[st].d[minid] + distance[minid][i] < cow[st].d[i]) {
				cow[st].d[i] = cow[st].d[minid] + distance[minid][i];
			}
		}
	}
}

int main(){
	scanf("%d %d %d", &n, &m, &k);
	for(int i = 1; i <= n; i++) {
		for(int j = 1; j <= i; j++) {
			if(i == j) {
				dis[i][j] = 0;
			} else {
				dis[i][j] = dis[j][i] = inf;
			}
		}
	}
	int x, y, z;
	for(int i = 1; i <= m; i++) {
		scanf("%d %d %d", &x, &y, &z);
		if(z < dis[x][y]) {
			dis[x][y] = z;	
		}
	}
	// 将所有边反向求距离
	for(int i = 1; i <= n; i++) {
		for(int j = 1; j <= n; j++) {
			dis1[i][j] = dis[j][i];
		}
	}
	dijkstra(dis, k);
	int ans = 0;
	for(int i = 1; i <= n; i++) {
		sum[i] += cow[k].d[i];
	}
	dijkstra(dis1, k);
	for(int i = 1; i <= n; i++) {
		sum[i] += cow[k].d[i];
	}
	// 注意要求总距离的最大值，所以要将来和去的值加起来再取max
	for(int i = 1; i <= n; i++) {
		ans = max(ans, sum[i]);
	}
	cout << ans;
	return 0;
}