个人训练赛第二十三场：寻路（弗洛伊德最短路）

问题 B: 寻路

题目描述
明明同学被困在一个荒凉的北极岛屿，他可以用小船乘着海流用1单位时间从一个岛移动到另一个岛。他得到了一个海洋地图，有N（1<=N<=100）条单向海流航线，编号为1…N。
告诉你他的起始位置M（1<=M<=N）和地图，请编程帮助明明确定到达每个岛的最短时间是多少。
输入为一个矩阵C，第r行，第c列的值若为1，则r到c存在海流，值为0则不存在海流。
输入
第1行：两个用空格隔开的整数：N和M
第2…N+1：第i+1行包含N个用空格隔开的整数：C_R
输出
第1…？？行：第一行输出M，第i+1行包含时刻i能到达的岛屿（升序排列）
样例输入 Copy
4 1
0 1 0 1
0 0 1 0
0 0 0 1
0 0 0 0
样例输出 Copy
1
2 4
3

思路：
这个题其实可以用很多办法解决，我的思路是跑弗洛伊德最短路
好吧其实是我只会弗洛伊德
定义一个二维数组dis来存储时间，由于要求最短的时间，所以给数组初始化一个很大的值。由一个岛到达另一个岛花费时间为1，所以我们循环去输入x，当x == 1时，就把这个时间存入dis数组，存完之后就是跑弗洛伊德最短路了。最后我们判断当前花费时间是否是初始的很大的值，如果不是，说明这条路就是我们要找的，把它存下来输出即可

int n,m,dis[105][105];
int main(){
     
    io >> n >> m;
    memset(dis,0x3f3f3f3f,sizeof dis); // 路径初始化 
    
    for(int i=1;i<=n;i++){
    	for(int j=1;j<=n;j++){
    		int x; io >> x;
    		if(x == 1) dis[i][j] = 1;
		}
	}
	
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			if(j == i) continue;
			for(int k=1;k<=n;k++){
				if(k == j || k == i) continue;
				dis[j][k] = min(dis[j][k],dis[j][i] + dis[i][k]);
				// 弗洛伊德求最短路 
			}
		}
	}
	
	dis[m][m] = 0; // 起点 
	vector<int>ans[102]; // 动态数组ans 
	for(int i=1;i<=n;i++){
		if(dis[m][i] != dis[0][0]){
			ans[dis[m][i]].push_back(i);
			// 如果不是无穷大，那么就把花费时间作为数组下标存储进来 
		}
	}
	for(int i=0;i<=n;i++){
		for(auto it:ans[i]){ // auto自动为变量匹配类型 
			printf("%d ",it);
		}
		puts("");
	}
    return 0;
}