7-15 周游世界分数 30（堆优化版dijkstra）

周游世界是件浪漫事，但规划旅行路线就不一定了…… 全世界有成千上万条航线、铁路线、大巴线，令人眼花缭乱。所以旅行社会选择部分运输公司组成联盟，每家公司提供一条线路，然后帮助客户规划由联盟内企业支持的旅行路线。本题就要求你帮旅行社实现一个自动规划路线的程序，使得对任何给定的起点和终点，可以找出最顺畅的路线。所谓“最顺畅”，首先是指中途经停站最少；如果经停站一样多，则取需要换乘线路次数最少的路线。

输入格式：

输入在第一行给出一个正整数N（≤100），即联盟公司的数量。接下来有N行，第i行（i=1,⋯,N）描述了第i家公司所提供的线路。格式为：

M S[1] S[2] ⋯ S[M]

其中M（≤100）是经停站的数量，S[i]（i=1,⋯,M）是经停站的编号（由4位0-9的数字组成）。这里假设每条线路都是简单的一条可以双向运行的链路，并且输入保证是按照正确的经停顺序给出的 —— 也就是说，任意一对相邻的S[i]和S[i+1]（i=1,⋯,M−1）之间都不存在其他经停站点。我们称相邻站点之间的线路为一个运营区间，每个运营区间只承包给一家公司。环线是有可能存在的，但不会不经停任何中间站点就从出发地回到出发地。当然，不同公司的线路是可能在某些站点有交叉的，这些站点就是客户的换乘点，我们假设任意换乘点涉及的不同公司的线路都不超过5条。

在描述了联盟线路之后，题目将给出一个正整数K（≤10），随后K行，每行给出一位客户的需求，即始发地的编号和目的地的编号，中间以一空格分隔。

输出格式：

处理每一位客户的需求。如果没有现成的线路可以使其到达目的地，就在一行中输出“Sorry, no line is available.”；如果目的地可达，则首先在一行中输出最顺畅路线的经停站数量（始发地和目的地不包括在内），然后按下列格式给出旅行路线：

Go by the line of company #X1 from S1 to S2.
Go by the line of company #X2 from S2 to S3.
......

其中Xi是线路承包公司的编号，Si是经停站的编号。但必须只输出始发地、换乘点和目的地，不能输出中间的经停站。题目保证满足要求的路线是唯一的。

输入样例：

4
7 1001 3212 1003 1204 1005 1306 7797
9 9988 2333 1204 2006 2005 2004 2003 2302 2001
13 3011 3812 3013 3001 1306 3003 2333 3066 3212 3008 2302 3010 3011
4 6666 8432 4011 1306
4
3011 3013
6666 2001
2004 3001
2222 6666

输出样例：

2
Go by the line of company #3 from 3011 to 3013.
10
Go by the line of company #4 from 6666 to 1306.
Go by the line of company #3 from 1306 to 2302.
Go by the line of company #2 from 2302 to 2001.
6
Go by the line of company #2 from 2004 to 1204.
Go by the line of company #1 from 1204 to 1306.
Go by the line of company #3 from 1306 to 3001.
Sorry, no line is available.

一眼dijkstra，需要维护两个花费，一个是中途经停站，一个是换乘次数。如何建图才能把这两个花费联系起来？对于每个公司的线路中的经停站到其他站的经停站数为下标相减，到其他站的换乘次数都是1 ，因此我们可以从前向后建边，边的权重其中一个为经停站数，换乘次数都为1，可以不用记录这个信息，另一个权重用来记录走的是哪个公司的线路（一开始我没记录这个，我是直接用一个二维数据记录点到点的公司，导致测试点2.3不通过）。

3
3 3 5 1
3 3 2 5
3 2 4 5
1
2 1
对于这个测试样例来说，从2到5经过的是二号公司的线路，我之前写的错误解法输出的公司是3.

for(int j=1;j<=m;j++) cin>>a[j];
        for(int j=1;j<=m;j++)
            for(int k=j+1;k<=m;k++)
            {
                int x=a[j],y=a[k];
                add(x,y,k-j);
                add(y,x,k-j);
                id[x][y]=i;
                id[y][x]=i;
            }

这是错误的记录方式。

以下是完整的正确代码

应该挺好理解的？

#include<bits/stdc++.h>
using namespace std;
const int N=10010,M=1000010,INF=0x3f3f3f3f;
typedef pair<int,string>PIS;
typedef pair<int,int>PII;
int n,k,m,bg,ed;
int a[110];
int h[N],e[M],ne[M],w1[M],w2[M],idx;
int dist1[N],dist2[N],pre[N],pre2[N];
bool st[N];

void add(int a,int b,int c,int d)
{
	e[idx]=b,ne[idx]=h[a],w1[idx]=c,w2[idx]=d,h[a]=idx++;
}
void dijkstra()
{
	memset(dist1,0x3f,sizeof dist1);
	memset(dist2,0x3f,sizeof dist2);
	dist1[bg]=0;
	dist2[bg]=0;
	memset(pre,-1,sizeof pre);
	memset(st,false,sizeof st);
	priority_queue<PII,vector<PII>,greater<PII>>q;
	q.push({dist1[bg],bg});
	while(q.size())
	{
		auto t=q.top();
		q.pop();
		int d=t.first,ver=t.second;
		if(st[ver]) continue;
		st[ver]=true;
		for(int i=h[ver];~i;i=ne[i])
		{
			int j=e[i];
			if(dist1[j]>d+w1[i])
			{
				dist1[j]=d+w1[i];
				dist2[j]=dist2[ver]+1;
				pre[j]=ver;
				pre2[j]=w2[i];
				q.push({dist1[j],j});
			}else if(dist1[j]==d+w1[i]&&dist2[j]>dist2[ver]+1)
			{
				dist2[j]=dist2[ver]+1;
				pre[j]=ver;
				pre2[j]=w2[i];
				q.push({dist1[j],j});
			}
		}
	}

}
void print(int u)
{
	if(u==bg) return;
	print(pre[u]);
	printf("Go by the line of company #%d from %04d to %04d.\n",pre2[u],pre[u],u);

}
void solve()
{
	cin>>n;
	memset(h,-1,sizeof h);
	for(int i=1;i<=n;i++)
	{
		cin>>m;
		for(int j=1;j<=m;j++) cin>>a[j];
		for(int j=1;j<=m;j++)
			for(int k=j+1;k<=m;k++)
			{
				int x=a[j],y=a[k];
				add(x,y,k-j,i);
				add(y,x,k-j,i);
			}
	}
	cin>>k;
	while(k--)
	{
		cin>>bg>>ed;
		dijkstra();
		if(dist1[ed]==INF) cout<<"Sorry, no line is available."<<endl;
		else
		{
			cout<<dist1[ed]<<endl;
			print(ed);

		}
	}
}
int main()
{
	int T=1;
	// cin>>T;
	while(T--) solve();
	return 0;
}