(L3-005)垃圾箱分布(dijkstra)

最新推荐文章于 2024-03-04 20:17:09 发布

原创最新推荐文章于 2024-03-04 20:17:09 发布 · 667 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#图论

最短路问题专栏收录该内容

32 篇文章

订阅专栏

该篇博客讨论了一道关于最短路径问题的算法题目，其中涉及到了Dijkstra算法的运用。题目要求在给定的居民点和垃圾箱之间找到最优的垃圾箱位置，使得所有居民到该垃圾箱的平均、最短和最长距离满足特定条件。博主通过分析题意，利用Dijkstra算法分别从每个垃圾箱出发计算最短路径，并找到了满足条件的垃圾箱编号及其距离信息。博客内容包括了算法实现细节和输入输出样例，适合对图论和算法感兴趣的读者学习。

题目链接：PTA | 程序设计类实验辅助教学平台

输入样例1：

输出样例1：

G1
2.0 3.3

输入样例2：

2 1 2 10
1 G1 9
2 G1 20

输出样例2：

No Solution

分析：这道题目考察的是最短路的知识，我们把垃圾箱看成是普通的点即可，只是把他们的编号设置在n+1~n+m，由于垃圾箱的数量不会很多，我们可以以每个垃圾箱为源点跑一边dijkstra，分别求出1~n号点到每个垃圾箱距离的最小值和最大值以及平均值，最后直接按照题目中要求的规则求一个答案即可。由于我们对垃圾箱跑dijkstra的顺序是从小到大的，所以我们前面选择的答案一定是编号较小的，所以我们没必要考虑编号的问题。

下面我来说一下输入，这个输入必须要严格控制格式，我们可以用string类型输入，而且我们还要判断每一个点是居民还是垃圾箱，如果是垃圾箱的话，我们只需要判断s串长度是2还是3即可判断垃圾箱编号，因为垃圾箱最多只有10个，所以比较简单，如果是居民，我们也只需要一个for循环就可以求出来编号，只是求出来垃圾箱编号别忘了加n。

最后想说的一点是，这道题目数据确实比较水，26分的数据居民的编号都是1位数，亲测有效

下面是代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
#include<algorithm>
#include<map>
#include<cmath>
#include<queue>
using namespace std;
const int N=1e5+10;
typedef pair<int,int>PII;
int h[N],e[N],ne[N],w[N],idx;
int d[N];
bool vis[N];
struct node{
	int mx;//垃圾箱所在位置到所有居民的最长距离 
	int mi;//垃圾箱所在位置到所有居民的最短距离 
	double dis;//垃圾箱所在位置到所有居民的平均距离
}p[13];
void add(int x,int y,int z)
{
	e[idx]=y;
	w[idx]=z;
	ne[idx]=h[x];
	h[x]=idx++;
}
void dijkstra(int x)
{
	memset(vis,false,sizeof vis);
	memset(d,0x3f,sizeof d);
	d[x]=0;
	priority_queue<PII,vector<PII>,greater<PII> >q;
	q.push({0,x});
	while(!q.empty())
	{
		int begin=q.top().second;
		q.pop();
		if(vis[begin]) continue;
		vis[begin]=true;
		for(int i=h[begin];i!=-1;i=ne[i])
		{
			int j=e[i];
			if(d[j]>d[begin]+w[i])
			{
				d[j]=d[begin]+w[i];
				q.push({d[j],j});
			}
		}
	}
}
int main()
{
	memset(h,-1,sizeof h);
	int n,m,k,D;
	cin>>n>>m>>k>>D;
	string s1,s2;
	for(int i=1;i<=k;i++)
	{
		int val;
		cin>>s1>>s2>>val;
		int u=0,v=0;
		if(s1[0]=='G')
		{
			if(s1.length()==2)
				u=s1[1]-'0'+n;
			else
				u=10+n;
		}
		else
		{
			for(int j=0;j<s1.length();j++)
				u=u*10+s1[j]-'0';
		}
		if(s2[0]=='G')
		{
			if(s2.length()==2)
				v=s2[1]-'0'+n;
			else
				v=10+n;
		}
		else
		{
			for(int j=0;j<s2.length();j++)
				v=v*10+s2[j]-'0';
		}
//		printf("%d %d\n",u,v);
		add(u,v,val);add(v,u,val);
	}
	for(int i=1;i<=m;i++)
	{
		dijkstra(i+n);
		p[i].dis=p[i].mi=p[i].mx=d[1];
		for(int j=2;j<=n;j++)
		{
			p[i].dis+=d[j];
			p[i].mi=min(p[i].mi,d[j]);
			p[i].mx=max(p[i].mx,d[j]);
		}
		p[i].dis=p[i].dis/n;
	}
	int i=1;
	while(p[i].mx>D&&i<=m) i++;
	if(i>m) printf("No Solution");
	else
	{
		int ans=i;
		for(int j=i+1;j<=m;j++)
		{
			if(p[j].mx>D) continue;
			if(p[j].mi>p[ans].mi)	ans=j;
			else if(p[j].mi==p[ans].mi)
			{
				if(p[j].dis<p[ans].dis) ans=j;
				//由于我们编号是从小到大遍历的，所以前面选的一定是编号较小的 
			}
		}
		printf("G%d\n",ans);
		printf("%.1lf %.1lf",(double)p[ans].mi,p[ans].dis);
	}
	return 0;
}