很想去的一家无人驾驶创业公司,技术面过了两面,面试官评价也都不错,败在了最后的coding面上。本来还存在侥幸的心理,但还是在生日这天收到了HR拒绝的电话。
很没出息的掉了几大滴眼泪,痛定思痛,把那天没做出来的题目又翻出来敲了一次,发现理顺了思路其实真的蛮简单的。被拒绝也是可以理解的吧。
上图是面试官出的题目,题目描述大致为:有一个人被困在最中心五角星标的位置,求最短的出去的路径是多少?每到达一个节点的花费是节点上的数字,出去是指达到白色的没有标数字的区域,迷宫层数不限。
一开始想用暴力搜索来求解,面试官让计算一下这种算法的时间复杂度,结果是n的指数型的。后来经过面试官提醒,想到了动态规划的求法,但是因为第一次用到嵌套vector,点之间的对应关系也很混乱,代码写得一塌糊涂,中间也多次想放弃。最后还是提交了个半成品上去,自己都不太想再看自己的代码。
今天重新梳理了一下思路:不考虑往回走(这里只是个假设,是最简单的情况,往回走的情况等以后再思索吧),从n层里回缩,记下与n-1层的每个节点相连的n层节点中最小的节点,一直回缩到第二层,也就是只有4个节点的层,这时候就可以比较四个元素的大小了,从而找到最小路径和。
程序输入为:第一个数为迷宫层数,然后从内到外输入迷宫的数字,每层按照顺时针顺序输入,从左上角开始。上图的输入应该为:
3 1 6 4 3 2 7 8 2 9 1 7 8 3
代码如下:
#include<iostream> #include<vector> #include<algorithm> using namespace std; vector< vector<int> > matrix; // 存储迷宫 vector<int> res; // 存储每次回缩的最小路径 int shortest; // 最小路径和(不包括最里层) void dg(int layer_num) { if (layer_num == 0) { // 第一次使用 std::min()函数, 定义在 #include<algorithm>里 shortest = min({ res[0], res[1], res[2], res[3] }); return; } int i = layer_num; for (int j = 0; j < 4 * i; j++) { int minX; if (j == 0) // 第一个点单独处理 { minX = min({ res[4 * (i + 1) - 1], res[0], res[1] }); res[j] = minX + matrix[i][j]; continue; } int x = (j - 1) / i; // 利用x从而简化n层与n-1层点的对应关系 if (j%i == 0) // 三个角(除去第一个点)所连的外层有三个点 { minX = min({ res[j + x], res[j + x + 1], res[j + x + 2] }); } else // 内点所连的外层有两个点 { minX = min({ res[j + x], res[j + x + 1] }); } res[j] = minX + matrix[i][j]; // 将n层的最小值回缩并加到第n-1层上 } dg(i - 1); //递归 } int main() { int layer_num; vector<int> layer; int ele; // 读取图数据 cin >> layer_num; for (int i = 0; i < layer_num; i++) { if (i == 0) { cin >> ele; layer.push_back(ele); matrix.push_back(layer); layer.clear(); continue; } for (int j = 0; j < 4 * i; j++) { cin >> ele; layer.push_back(ele); if (i == layer_num - 1) { res.push_back(ele); // 一开始的res初始为最外层的一圈数字 } } matrix.push_back(layer); layer.clear(); } dg(matrix.size() - 2); cout << shortest + matrix[0][0]<< endl; return 0; }