L3-022 地铁一日游 (30 分)【floyd+dfs】

森森喜欢坐地铁。这个假期，他终于来到了传说中的地铁之城——魔都，打算好好过一把坐地铁的瘾！魔都地铁的计价规则是：起步价 2 元，出发站与到达站的最短距离（即计费距离）每 K 公里增加 1 元车费。

例如取 K = 10，动安寺站离魔都绿桥站为 40 公里，则车费为 2 + 4 = 6 元。

为了获得最大的满足感，森森决定用以下的方式坐地铁：在某一站上车（不妨设为地铁站 A），则对于所有车费相同的到达站，森森只会在计费距离最远的站或线路末端站点出站，然后用森森美图 App 在站点外拍一张认证照，再按同样的方式前往下一个站点。坐着坐着，森森突然好奇起来：在给定出发站的情况下（在出发时森森也会拍一张照），他的整个旅程中能够留下哪些站点的认证照？地铁是铁路运输的一种形式，指在地下运行为主的城市轨道交通系统。一般来说，地铁由若干个站点组成，并有多条不同的线路双向行驶，可类比公交车，当两条或更多条线路经过同一个站点时，可进行换乘，更换自己所乘坐的线路。举例来说，魔都 1 号线和 2 号线都经过人民广场站，则乘坐 1 号线到达人民广场时就可以换乘到 2 号线前往 2 号线的各个站点。换乘不需出站（也拍不到认证照），因此森森乘坐地铁时换乘不受限制。

输入格式:

输入第一行是三个正整数 N、M 和 K，表示魔都地铁有 N 个车站 (1 ≤ N ≤ 200)，M 条线路 (1 ≤ M ≤ 1500)，最短距离每超过 K 公里 (1 ≤ K ≤ 106)，加 1 元车费。

接下来 M 行，每行由以下格式组成：

<站点1><空格><距离><空格><站点2><空格><距离><空格><站点3> ... <站点X-1><空格><距离><空格><站点X>

其中站点是一个 1 到 N 的编号；两个站点编号之间的距离指两个站在该线路上的距离。两站之间距离是一个不大于 106 的正整数。一条线路上的站点互不相同。

注意：两个站之间可能有多条直接连接的线路，且距离不一定相等。

再接下来有一个正整数 Q (1 ≤ Q ≤ 200)，表示森森尝试从 Q 个站点出发。

最后有 Q 行，每行一个正整数 **Xi**，表示森森尝试从编号为 Xi 的站点出发。

输出格式:

对于森森每个尝试的站点，输出一行若干个整数，表示能够到达的站点编号。站点编号从小到大排序。

输入样例:

6 2 6
1 6 2 4 3 1 4
5 6 2 6 6
4
2
3
4
5

输出样例:

1 2 4 5 6
1 2 3 4 5 6
1 2 4 5 6
1 2 4 5 6

 这题我感觉题意备受争议，下面是我的思路：

1.用floyd算法求出各个结点到各个结点的距离。

2.找出一个结点到各个结点的最小花费的最远结点。

3.用st存储一个结点到各个结点的最小花费的最远结点，这样后续dfs遍历的时候，一个结点到下一个结点一定是能到达的最小花费的最远结点。

4.询问一个结点时，用dfs遍历一遍，能到达的结点就是其答案。

（踩的坑点：一条线路的端点，可用ed[Max]数组标记，mp[j].size()==0||mp[j].size()==1不可判断是否是一条线路的端点，因为一个点可能是多条线路的端点）

AC代码：

#include<bits/stdc++.h>
#define pb push_back
#define ll long long
using namespace std;
const int Max=1e6+5;
const int INF=1e9+5;
int n,m,k;
struct node{
	int to;
	ll key;
};
vector<node>mp[Max];
ll room[205][205];
vector<int>st[Max];
void floyd(){
	for(int k=1;k<=n;k++){
		for(int i=1;i<=n;i++){
			for(int j=1;j<=n;j++){
				if(i!=j) room[i][j]=min(room[i][j],room[i][k]+room[k][j]);
			}
		}
	}
}
bool vis[Max];
void dfs(int x){
	for(int i=0;i<st[x].size();i++){
		if(!vis[st[x][i]]){
			vis[st[x][i]]=true;
			dfs(st[x][i]);
		}
	}
}
bool ed[Max];
int main(){
	cin>>n>>m>>k;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++) room[i][j]=INF;
	}
	for(int i=0;i<m;i++){
		int pre;cin>>pre;
		int w,x;ed[pre]=true;
		while(cin>>w>>x){
			mp[pre].pb({x,w});
			mp[x].pb({pre,w});
			
			if(room[pre][x]>w) room[pre][x]=w;
			if(room[x][pre]>w) room[x][pre]=w;
			pre=x;
			if(getchar()=='\n') break;
		}
		ed[pre]=true;
	}
	floyd();
	for(int i=1;i<=n;i++){
		map<int,ll>dp;
		for(int j=1;j<=n;++j){
			if(room[i][j]==INF) continue;
			dp[room[i][j]/k+2]=max(room[i][j],dp[room[i][j]/k+2]);
		}
		for(int j=1;j<=n;j++){
			if(dp[room[i][j]/k+2]==room[i][j]||i!=j&&room[i][j]!=INF&&ed[j]){
				st[i].pb(j);
			}
		}
	}
	int q;cin>>q;
	while(q--){
		int x;cin>>x;
		for(int i=1;i<=n;i++) vis[i]=false;
		vis[x]=true;
		dfs(x);
		set<int>s;
		for(int i=1;i<=n;i++){
			if(vis[i]) s.insert(i);
		}
		vector<int>temp;
		for(set<int>::iterator it=s.begin();it!=s.end();it++){
			int hy=*it;
			temp.pb(hy);
		}
		for(int i=0;i<temp.size()-1;i++) cout<<temp[i]<<' ';cout<<temp[temp.size()-1]<<endl;
	}
}