PAT——1131 Subway Map 甲级

题目

https://pintia.cn/problem-sets/994805342720868352/problems/994805347523346432

题意

给出n条地铁线路，要求对于给出的起始点输出最短路径中的中转站。最短路径的意思是途经的站点最少，站点数量相同则中转站最少

代码解析

通过高四位低四位的形式在line中存储每两站的连线所在的线路，并利用dfs和回溯法从起点开始遍历线路。留下符合要求的最短线路path后，遍历path，通过line判断每两站之间的线路是否发生改变，如果改变必是上一站经过了中转站，输出即可

注意：即使最后一站不是中转站，终点和上一个中转站也要输出

AC代码

#include<bits/stdc++.h>
using namespace std;
int vis[10000],minCnt,minTran,start,last;
unordered_map<int,int> line;
unordered_map<int,vector<int>> v;
vector<int> path,tmpPath;

int cntTran(vector<int> a)
{
	int cnt=-1,preline=0;
	for(int i=1;i<a.size();i++)
	{
		if(line[a[i-1]*10000+a[i]]!=preline) cnt++;
		preline=line[a[i-1]*10000+a[i]];
	}
	return cnt;
}

void dfs(int root,int cnt)
{
	if(root==last && (cnt<minCnt|| (cnt==minCnt&&cntTran(tmpPath)<minTran) ) )
	{
		minCnt=cnt;
		minTran=cntTran(tmpPath);
		path=tmpPath;
	}
	if(root==last) return;
	for(int i=0;i<v[root].size();i++)
	{
		if(vis[v[root][i]]==0)
		{
			vis[v[root][i]]=1;
			tmpPath.push_back(v[root][i]);
			dfs(v[root][i],cnt+1);
			vis[v[root][i]]=0;
			tmpPath.pop_back();
		}
	}
}
int main()
{
	int n,m,k,pre,tmp;
	scanf("%d",&n);
	for(int i=0;i<n;i++)
	{
		scanf("%d %d",&m,&pre);
		for(int j=1;j<m;j++)
		{
			scanf("%d",&tmp);
			v[pre].push_back(tmp);
			v[tmp].push_back(pre);
			line[pre*10000+tmp]=line[tmp*10000+pre]=i+1;
			pre=tmp;
		}
	}
	scanf("%d",&k);
	while(k--)
	{
		minCnt=minTran=INT_MAX;
		scanf("%d %d",&start,&last);
		tmpPath.clear();
		tmpPath.push_back(start);
		vis[start]=1;
		dfs(start,0);
		vis[start]=0;
		printf("%d\n",minCnt);
		int preline=0,preTran=start;
		for(int i=1;i<path.size();i++)
		{
			if(line[path[i-1]*10000+path[i]]!=preline)
			{
				if(preline) printf("Take Line#%d from %04d to %04d.\n",preline,preTran,path[i-1]);
				preline=line[path[i-1]*10000+path[i]];
				preTran=path[i-1];
			}
		}
		if(preline) printf("Take Line#%d from %04d to %04d.\n",preline,preTran,last);
	}
}

参考（推荐）

PAT 1131. Subway Map (30) -甲级