寒假训练POJ-3268题解

问题链接：POJ-3268

思路

一开始以为就是用多次最短路来求解，把各个奶牛到终点的最短距离算出来，再把终点到各个奶牛的最短距离算出来，计算每头奶牛所用最短时间，输出最大的，结果超时了，看了别人的题解才恍然大悟。因为前者需要运行多次最短路，后者只用一次就可以解决问题，由于前者的原因超时了。后者不变，需要改变求前者时间的办法，因为是单向的路径，所以如果把路径全部反过来，再以终点作为起点来计算到各个奶牛的最短路其实也是一样的，这样只运行了一次最短路，不会超时。

AC的代码

#include<iostream>
#include<algorithm>
using namespace std;
int cost[1005][1005], cost2[1005][1005];
int visit[1005];
int d[1005], d2[1005];
int n;
int min(int, int);
void f(int);
void f2(int);
int main()
{
	int m, z;
	cin >> n >> m >> z;
	for (int i = 0; i < m; i++)
	{
		int a, b, c;
		cin >> a >> b >> c;
		cost[a][b] = c;
		cost2[b][a] = c;
	}
	f(z);
	f2(z);
	int p = 0;
	for (int i = 1; i <= n; i++)
		if (d[i] + d2[i] > p) p = d[i] + d2[i];
	cout << p << endl;
	return 0;
}
int min(int x, int y)
{
	if (x <= y) return x;
	else return y;
}
void f(int x)
{
	fill(d, d + n + 1, 99999);
	fill(visit, visit + n + 1, 0);
	d[x] = 0;
	while (true)
	{
		int k = -1;
		for (int i = 1; i <= n; i++)
			if (!visit[i] && (k == -1 || d[i] < d[k])) k = i;
		if (k == -1) break;
		visit[k] = 1;
		for (int i = 1; i <= n; i++)
		{
			if (cost[k][i] == 0) cost[k][i] = 99999;
			d[i] = min(d[i], d[k] + cost[k][i]);
		}
	}
}
void f2(int x)
{
	fill(d2, d2 + n + 1, 99999);
	fill(visit, visit + n + 1, 0);
	d2[x] = 0;
	while (true)
	{
		int k = -1;
		for (int i = 1; i <= n; i++)
			if (!visit[i] && (k == -1 || d2[i] < d2[k])) k = i;
		if (k == -1) break;
		visit[k] = 1;
		for (int i = 1; i <= n; i++)
		{
			if (cost2[k][i] == 0) cost2[k][i] = 99999;
			d2[i] = min(d2[i], d2[k] + cost2[k][i]);
		}
	}
}