L3-005 垃圾箱分布

大家倒垃圾的时候，都希望垃圾箱距离自己比较近，但是谁都不愿意守着垃圾箱住。所以垃圾箱的位置必须选在到所有居民点的最短距离最长的地方，同时还要保证每个居民点都在距离它一个不太远的范围内。

现给定一个居民区的地图，以及若干垃圾箱的候选地点，请你推荐最合适的地点。如果解不唯一，则输出到所有居民点的平均距离最短的那个解。如果这样的解还是不唯一，则输出编号最小的地点。

输入格式：

输入第一行给出4个正整数：N（<= 103）是居民点的个数；M（<= 10）是垃圾箱候选地点的个数；K（<= 104）是居民点和垃圾箱候选地点之间的道路的条数；DS是居民点与垃圾箱之间不能超过的最大距离。所有的居民点从1到N编号，所有的垃圾箱候选地点从G1到GM编号。

随后K行，每行按下列格式描述一条道路：
P1 P2 Dist
其中P1和P2是道路两端点的编号，端点可以是居民点，也可以是垃圾箱候选点。Dist是道路的长度，是一个正整数。

输出格式：

首先在第一行输出最佳候选地点的编号。然后在第二行输出该地点到所有居民点的最小距离和平均距离。数字间以空格分隔，保留小数点后1位。如果解不存在，则输出“No Solution”。

输入样例1：
4 3 11 5
1 2 2
1 4 2
1 G1 4
1 G2 3
2 3 2
2 G2 1
3 4 2
3 G3 2
4 G1 3
G2 G1 1
G3 G2 2
输出样例1：
G1
2.0 3.3
输入样例2：
2 1 2 10
1 G1 9
2 G1 20
输出样例2：
No Solution

---

把每个垃圾桶拎出来求最短路
垃圾桶的序号设立为 N+1 ~ N+M

最后一个点运行时错误是为什么？ 数组越界？

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;

const int maxn = 1500;
const int INF = 0x3f3f3f3f;
int dist[maxn];
int vis[maxn];
int N, M, K, Dmax; // 居民数，垃圾箱数, 路径数，垃圾箱最大距离
vector<pair<int, int> >E[maxn];
struct Ans
{
    int mindis;
    double avedis;
    string id;
    Ans(int a, double b, string c) {
        mindis = a;
        avedis = b;
        id = c;
    }
    bool operator<(const Ans &a) const {
        return (mindis > a.mindis) || (mindis == a.mindis && avedis < a.avedis) || (mindis==a.mindis && avedis == a.avedis && id < a.id);
    }
};
vector<Ans> ans;

int getid(char g[])
{
    int x;
    if (g[0] == 'G')
    {
        if (g[2] != '\0')
            x = (g[1] - '0') * 10 + g[2] - '0';
        else
            x = g[1] - '0';
        x += N;
    }
    else
    {
        if (g[1] != '\0'){
            x = (g[0] - '0') * 10 + g[1] - '0';
        }
        else
            x = g[0] - '0';
    }
    return x;
}


void dijkstra(int sta) // sta：起点（当前的垃圾箱）
{
    priority_queue<pair<int, int> >Q;
    dist[sta] = 0;
    Q.push({-dist[sta], sta});
    while (!Q.empty())
    {
        int head = Q.top().second;
        Q.pop();
        if (vis[head]) continue;
        vis[head] = 1;
        for (int i = 0; i < E[head].size(); i ++)
        {
            int v = E[head][i].first;
            int d = E[head][i].second;
            if (dist[v] > d + dist[head])
            {
                dist[v] = d + dist[head];
                Q.push({-dist[v], v});
            }
        }
    }
}

int main()
{
    // freopen("/Users/zhbink/Documents/C++/C++/in.txt", "r", stdin);
    scanf("%d%d%d%d", &N, &M, &K, &Dmax);
    for (int i = 0; i < K; i++)
    {
        int a, b, c;
        char g1[3], g2[3];
        scanf("%s%s%d", g1, g2, &c);
        a = getid(g1);
        b = getid(g2);
        E[a].push_back({b, c});
        E[b].push_back({a, c});
    }
    for (int i = N+1; i <= N+M; i ++)
    {
        for (int j = 1; j <= N+M; j ++) dist[j] = INF;
        memset(vis, 0, sizeof(vis));
        dijkstra(i);
        int mindis = INF;
        int maxdis = -1;
        double dissum = 0;
        for (int j = 1; j <= N; j ++)
        {
            if (dist[j] < mindis) mindis = dist[j];
            if (dist[j] > maxdis) maxdis = dist[j];
            dissum += (double)dist[j];
        }
        if (maxdis > Dmax) continue;
        string id = "G";
        id += to_string(i-N);
        ans.push_back(Ans(mindis, dissum/N, id));
    }
    if (ans.size())
    {
        sort(ans.begin(), ans.end());
        printf("%s\n%.1f %.1f\n", ans[0].id.c_str(), (double)ans[0].mindis, (int)(ans[0].avedis*10+0.5)/10.0);
    }
    else printf("No Solution\n");
    return 0;
}