PTA团体程序设计天梯赛GPLT L3-005. 垃圾箱分布(Dijkstra)

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

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

输入格式:

输入第一行给出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

#include 
using namespace std;
int n,m,k,d,e[1200][1200],dis[1200],vis[1200];
double ansmdis = -1,ansadis = 9999999,ansid = -1;
int main(){
    cin>>n>>m>>k>>d;
    fill(e[0],e[0] + 1200*1200,9999999);
    for(int i = 0;i < k;i++){
        string s1,s2;
        int c,a,b;
        cin>>s1>>s2>>c;
        if(s1[0] == 'G'){
            a = n + stoi(s1.substr(1));
        }else{
            a = stoi(s1);
        }
        if(s2[0] == 'G'){
            b = n + stoi(s2.substr(1));
        }else{
            b = stoi(s2);
        }
        e[a][b] = c;
        e[b][a] = c;
    }
    for(int index = n+1;index <= n+m;index++){
        fill(dis,dis+1200,9999999);
        fill(vis,vis+1200,0);
        double mindis = 9999999,avrdis = 0;
        dis[index] = 0;
        for(int i = 0;i < n + m;i++) {
            int minn = 9999999,u = -1;
            for (int j = 1; j <= n + m; j++) {
                if (dis[j] < minn && vis[j] != 1) {
                    u = j;
                    minn = dis[j];
                }
            }
            if(u == -1) break;
            vis[u] = 1;
            for (int v = 1; v <= n + m; v++) {
                if(dis[v] > dis[u] + e[u][v] && vis[v] != 1){
                    dis[v] = dis[u] + e[u][v];
                }
            }
        }
        for(int i = 1;i <= n;i++){
            if(dis[i] > d){
                mindis = -1;
                break;
            }
            if(dis[i] < mindis){
                mindis = dis[i];
            }
            avrdis += dis[i];
        }
        avrdis /= n;
        if(mindis == -1){
            continue ;
        }
        else{
            if(mindis > ansmdis){
                ansid = index;
                ansadis = avrdis;
                ansmdis = mindis;
            }else if(mindis == ansmdis && avrdis < ansadis) {
                ansid = index;
                ansadis = avrdis;
            }
        }
    }
    if(ansid == -1)
        printf("No Solution");
    else {
        cout<<"G"<<ansid-n<<endl;
        printf("%.1f %.1f", ansmdis, ansadis);
    }
	return 0;
}

你可能感兴趣的:(PTA练习集,dijkstra)