[PAT解题报告] Gas Station
麻烦题,给定n个房间编号1-N, m个加油站,编号(N+1)-(N+m),
若干条路。加油站要合法,需要满足加油站到每个房间的距离都不超过一个给定的值D。
在合法的加油站里选出一个加油站,要求它到最近房间的距离尽可能小,如果相等,再让它到所有房间的平均距离尽可能小,如果还相等,要加油站的编号尽可能小。
分析: 只要加油站合法,选加油站就容易了。加油站合法不合法主要看加油站到每个房间的距离,所以对每个加油站来说就是个单源图的最短路。
我采用的是优先队列的办法算最短路。
大概框架如下:
优先队列里存 (点,距离)
先把(起点,0)入队。
每个点只有第一次出队是有效的,因为第一次出队时距离最小,我们允许重复入队和出队,就是重复出队时直接忽略就可以了。
然后扩展也很简单,当前点x第一次出队的时候,距离d[x],是起点到x的最短距离,加一条边(x, y), 把(y, d[x] +
dist(x, y))入队,dist(x, y)表示(x,y)那条边的长度……
其实就是用优先队列做bfs而已。
代码:
#include <cstdio> #include <queue> #include <vector> #include <string> #include <cstring> using namespace std; priority_queue<pair<int,int> > q,qq; int n,D; vector<pair<int,int> > con[1024]; bool mark[1024]; int d[1024]; char s[100]; int get() { int x; if (s[0] == 'G') { sscanf(s + 1, "%d", &x); return n + x - 1; } sscanf(s, "%d", &x); return x - 1; } void test(int from) { q = qq; q.push(make_pair(0, from)); memset(mark,0,sizeof(mark)); memset(d, 0xff,sizeof(d)); while (!q.empty()) { int at = q.top().second; int dis = q.top().first; q.pop(); if (mark[at]) { continue; } mark[at] = true; d[at] = -dis; if ((at < n) && (d[at] > D)) { return; } for (int i = 0; i < con[at].size(); ++i) { q.push(make_pair(dis - con[at][i].second, con[at][i].first)); } } } int main() { int m, e; for (scanf("%d%d%d%d",&n,&m,&e,&D);e;--e) { scanf("%s",s); int x = get(), z; scanf("%s%d",s,&z); int y = get(); con[x].push_back(make_pair(y, z)); con[y].push_back(make_pair(x, z)); } int besttotal = 2000000000, bestmin = -1, ind = -1; for (int i = 0; i < m; ++i) { test(n + i); bool mark = true; int maytotal = 0; int maymin = 2000000000; for (int j = 0; j < n; ++j) { if ((d[j] > D) || (d[j] < 0)) { mark = false; break; } maytotal += d[j]; maymin = min(maymin, d[j]); } if (mark && ((ind < 0) || (bestmin < maymin) || ((bestmin == maymin) && (besttotal > maytotal)))) { ind = i; bestmin = maymin; besttotal = maytotal; } } if (ind >= 0) { printf("G%d\n%.1lf %.1lf\n",ind + 1, bestmin * 1., besttotal * 1. / n); } else { puts("No Solution"); } return 0; }
原题链接: http://www.patest.cn/contests/pat-a-practise/1072