2019阿里校招测评题(编程)

明小学的小朋友们要举行一年一度的接力跑大赛了，但是小朋友们却遇到了一个难题：设计接力跑大赛的线路，你能帮助他们完成这项工作么？
光明小学可以抽象成一张有N个节点的图，每两点间都有一条道路相连。光明小学的每个班都有M个学生，所以你要为他们设计出一条恰好经过M条边的路径。
光明小学的小朋友们希望全盘考虑所有的因素，所以你需要把任意两点间经过M条边的最短路径的距离输出出来以供参考。

你需要设计这样一个函数：
res[][] Solve( N, M, map[][]);
注意：map必然是N * N的二维数组，且map[i][j] == map[j][i]，map[i][i] == 0，-1e8 <= map[i][j] <= 1e8。（道路全部是无向边，无自环）2 <= N <= 100, 2 <= M <= 1e6。

map数组表示了一张稠密图，其中任意两个不同节点i,j间都有一条边，边的长度为map[i][j]。N表示其中的节点数。
你要返回的数组也必然是一个N * N的二维数组，表示从i出发走到j，经过M条边的最短路径
你的路径中应考虑包含重复边的情况。

样例：

N = 3
M = 2
map = {
 {0, 2, 3},
 {2, 0, 1},
 {3, 1, 0}
}

输出结果result为：
result = {
 {4, 4, 3},
 {4, 2, 5},
 {3, 5, 2}
}

输入样例：

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

输出样例：

[[4, 4, 3],

[4, 2, 5],

[3, 5, 2]]

 

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

vector<vector<int>> result;
vector<vector<int>> map;
int mindis = INT_MAX;

void dfs(int start, int end, int N, int M, int distance, int steps)
{
	if (steps == M)
	{
		if (start == end && distance < mindis)
			mindis = distance;
		return;
	}
	int rdis;
	for (int next = 0; next < N; next++)
	{
		if (next == start)
			continue;
		rdis = distance + map[start][next];
		dfs(next, end, N, M, rdis, steps + 1);
	}
}
vector<vector<int>> solution(int N, int M)
{
	result = vector<vector<int>>(N, vector<int>(N, 0));
	int dis = 0;
	for (int i = 0; i < N; i++)
	{
		for (int j = 0; j < N; j++)
		{
			dfs(i, j, N, M, 0, 0);
			result[i][j] = mindis;
			mindis = INT_MAX;
		}
	}
	return result;
}
int main()
{
	int N, M;
	cin >> N >> M;
	int temp;
	map = vector<vector<int>>(N, vector<int>(N, 0));
	result=vector<vector<int>>(N, vector<int>(N, 0));
	for (int i = 0; i < N; i++)
	{
		for (int j = 0; j < N; j++)
		{
			cin >> temp;
			map[i][j] = temp;
		}
	}
	solution(N, M);
	for (int i = 0; i < result.size(); i++)
	{
		for (int j = 0; j < result[i].size(); j++)
			cout << result[i][j] << " ";
		cout << endl;
	}
	return 0;
}

参考博客：https://blog.csdn.net/u012465304/article/details/81180707