代码源div2 每日一题 dijkstra

最新推荐文章于 2024-10-04 17:07:53 发布

crazycarryai

最新推荐文章于 2024-10-04 17:07:53 发布

阅读量124

点赞数

分类专栏：每日一题文章标签： c++ 图论

本文链接：https://blog.csdn.net/qq_44647370/article/details/124600030

版权

每日一题专栏收录该内容

9 篇文章 0 订阅

订阅专栏

本文介绍了如何利用图的邻接表结构和Dijkstra算法解决一个问题：在给定的一棵树中，判断两个节点是在同一层相遇还是在路径上相遇。通过计算两个节点到根节点的距离和，若为偶数则在村镇相遇，否则在路上相遇。代码实现采用C++，并讨论了BFS作为替代解法的可能性。

摘要由CSDN通过智能技术生成

Collision

把该图类比成树，转化成求两个节点在树的第几层，根节点随意取，这里取1为根节点，将两个节点的层数相加看奇偶，是偶数则能在村镇相遇，否则只能在路上相遇

用邻接表存图，因为无向图，需开到2e5的大小，稀疏图跑堆优化的djikstra

其实bfs也可以做此题

#include<iostream>
#include<cstring>
#include<queue>

using namespace std;

const int N=2e5+5;

typedef pair<int,int> PII;

int n,q,idex,nex[N],e[N],h[N],dis[N],vis[N];

void add(int a,int b)
{
	e[idex]=b,nex[idex]=h[a],h[a]=idex++;
}

void dijkstra()
{
	memset(dis,0x3f,sizeof(dis));
	dis[1]=0;
	priority_queue<PII,vector<PII>,greater<PII>> heap;
	heap.push({0,1});
	while(heap.size()!=0)
	{
		int t=heap.top().second;
		heap.pop();
		if(vis[t]==1) continue;
		vis[t]=1;
		for(int i=h[t];i!=-1;i=nex[i])
		{
			int j=e[i];
			if(dis[j]>dis[t]+1)
			{
				dis[j]=dis[t]+1;
				heap.push({dis[j],j});
			}
		}
	}
}

int main()
{
	scanf("%d%d",&n,&q);
	int a,b;
	memset(h,-1,sizeof(h));
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&a,&b);
		add(a,b);
		add(b,a);
	}
	dijkstra();
	for(int i=1;i<=q;i++)
	{
		scanf("%d%d",&a,&b);
		if((dis[a]+dis[b])%2==0)
		{
			puts("Town");
		}else
		{
			puts("Road");
		}
	}
}