2022牛客多校3 J-Journey

 

示例1

输入

4
3 4 0 0
0 0 4 3
2 1 0 0
2 0 0 1
4 2 4 1

输出

1

说明

One of the optimal route is: turn right at crossroad 2, turn right at crossroad 3, turn right at crossroad 1, and turn around at crossroad 4.

The picture is as follows:

 

题目大意：给定n个十字路口，一个路口最多可以从4个方向到达，当从一个路口通过左转，直行，掉头到达另一个路口时需要等一次红灯，右转则不需要等红灯，问至少等几次红灯可以从起始路口到达终点。

解题思路：我们可以把是否需要等红灯抽象成一条边，如果需要等这条边权值为1，不需要等权值为0，然后跑一边dijkstra即可，这个题的主要难点在于如何唯一表示一条路，以及这个题的建图比较绕，好多都是反着来的，具体可以看代码。

代码如下：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <unordered_map>
#include <bits/stdc++.h> 
using namespace std;
const int N=5e5+10;
int n,dist[N][4],cross[N][4],st1,st2,ed1,ed2;
unordered_map<int,int>id[N];
bool vis[N][4];
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		for(int j=0;j<4;j++)
		{
			 cin>>cross[i][j];//记录从j方向可以走到i点的点，由题目可知这里输入的点是起点 
			 id[i][cross[i][j]]=j;//记录从cross[i][j]走到i点的方向 
		}
	}
	cin>>st1>>st2>>ed1>>ed2;
	memset(dist,0x3f3f3f3f,sizeof dist);
	int idx=id[st2][st1];//记录从st1到st2的方向 
	dist[st2][idx]=0;
	priority_queue<tuple<int,int,int>,vector< tuple<int,int,int> >,greater< tuple<int,int,int> > >q;
	q.push({0,st2,idx});//第一个记录距离起誓道路的距离，第二个记录到达的当前点，第3个记录从哪个方向到达的当前点 
	while(q.size())
	{
		auto [d,u,x]=q.top();
		q.pop();
		if(vis[u][x])//用一个方向和到达的点可以唯一标识一条道路，该数组用来记录这条路是否已经走过 
		continue;
		vis[u][x]=true;
		for(int i=0;i<4;i++)
		{
			int de=(i!=(x+1)%4);//判断当前方向是否是右转，如果不是右转就需要等一次红灯 
			int v=cross[u][i];//v表示能从i方向到达当前点的点 
			if(!v)
			int e=id[v][u];//从当前点到v点的方向 
			if(d+de<dist[v][id[v][u]])
			{
				dist[v][id[v][u]]=d+de;
				q.push({dist[v][id[v][u]],v,id[v][u]});
			}
		}
	 } 
	 int ans=dist[ed2][id[ed2][ed1]];
	 if(ans==0x3f3f3f3f)
	 ans=-1;
	 cout<<ans<<endl;
	 return 0;
}