PAT A 1072 Gas Station (30分)

一、题目概述

给定N个居民区和M个加油站备选址，即各点间的距离。在M个油站备选址中找到最合适的一处。

油站选址标准：
1、N个居民区都在服务范围内。
2、距最短距离尽可能的远。
3、若有另一油站距居民区最短距离相同，选择距离居民区平均距离最短的。

二、思路

1、关于图的建立
（1）存储为邻接矩阵无向图，居民区下标1~N，油站下标N + 1 ~ N + M;
（2）用string读入居民区或油站的下标，若str以G开头，取G后面的数字再加N（小心G10时，G后是两位数），是油站下标；若str不以G开头，仅转换为数字，是居民区下标。

2、关于距离计算
从N + 1到N + M调用M次dijkstra算法，求单源最短路径，并统计最大距离和最小距离。
若最大距离超过Ds，说明有居民区不在服务范围，排除；
然后依次比较，找到最终油站位置。

三、代码

#include <iostream>
#include <cstdlib>
#include <vector>
#include <algorithm>
using namespace std;
#define INF 0x7FFFFFFF
int main()
{
	int N, M, K, Ds, ans_i = 0, ans_d = 0;
	double ans_avg;
	scanf("%d %d %d %d", &N, &M, &K, &Ds);
	vector<vector<int>> G(N + M + 1, vector<int>(N + M + 1, INF));
	for( int i = 0, v1, v2, d; i < K; ++i )
	{
		char str1[5], str2[5];
		scanf("%s %s %d", str1, str2, &d);
		v1 = str1[0] == 'G' ? (N + atoi(str1 + 1)):atoi(str1);
		v2 = str2[0] == 'G' ? (N + atoi(str2 + 1)):atoi(str2);
		G[v1][v2] = G[v2][v1] = d;
	}
	for( int i = N + 1, min_len, max_len; i <= N + M; ++i )
	{
		vector<int> dist(N + M + 1, INF), vset(N + M + 1, 0);
		dist[i] = 0;
		min_len = INF, max_len = 0;
		double avg = 0;
		for( int j = 0, Min, v; j < N + M; ++j )
		{
			Min = INF;
			for( int k = 1; k <= N + M; ++k)
				if( !vset[k] && dist[k] < Min )
				{
					v = k;
					Min = dist[k];
				}
			if( Min == INF )
				break;
			vset[v] = 1;
			for( int k = 1; k <= N + M; ++k )
				if( G[v][k] != INF && dist[v] + G[v][k] < dist[k] )
					dist[k] = dist[v] + G[v][k];
		}
		for( int j = 1; j <= N; ++j )
		{
			avg += dist[j];
			min_len = min(min_len, dist[j]);
			max_len = max(max_len, dist[j]);
		}
		avg /= N;
		if( max_len <= Ds && ( min_len > ans_d || ( min_len == ans_d && avg < ans_avg ) ) )
		{
			ans_i = i - N;
			ans_d = min_len;
			ans_avg = avg;
		}
	}
	if( ans_i )
		printf("G%d\n%d.0 %.1f", ans_i, ans_d, ans_avg);
	else printf("No Solution");
}