这篇文章将为大家详细讲解有关字节的笔试题和算法题示例分析,文章内容质量较高,因此小编分享给大家做个参考,希望大家阅读完这篇文章后对相关知识有一定的了解。
题目给定一个包含 m x n 个元素的矩阵(m 行, n 列),请按照顺时针螺旋顺序,返回矩阵中的所有元素。
输入:
[ [ 1, 2, 3 ], [ 4, 5, 6 ], [ 7, 8, 9 ] ]
输出:
[1,2,3,6,9,8,7,4,5]
思路
基本上围绕的思路就是:一层层向里处理,按顺时针依次遍历:上、右、下、左。
其实很类似于迷宫的走法,走到格子后,判断下一个格子,还能不能走,也就是边界条件。遇到边界条件后,顺着上面的顺序: 上、右、下、左。
所以我们可以有几个约束条件:
是不是在这个范围内,不能超过这些范围。
这个格子是不是走过,存一下之前的状态。
记录当前方向,抱着下一个方向是对的。
深度优先遍历
按照深度优先遍历思路来写,我们可以构造常见的dfs模版:
const spiralOrder = function (matrix) { if (matrix.length === 0) return []; const result = [], dx = [0, 1, 0, -1], dy = [1, 0, -1, 0], col = matrix.length, row = matrix[0].length; // isCheckMatrix记录是否走过 const isCheckMatrix = Array.from(new Array(col), () => (new Array(row).fill(false))) const dfs = (x, y, directionIndex) => { // 逻辑代码 // 通常这里做逻辑处理,边界处理 } }; dfs(0, 0, 0); return result };
这应该就是基础的模版,唯一不同的是,我们看dfs的三个参数,x,y,directionIndex。
x和y参数很好理解,这个directionIndex参数含义就是告诉我们当前前进的方向。
接下来,是写我们的逻辑部分。首先确定接下来走到哪一个格子:
dx = [0, 1, 0, -1] dy = [1, 0, -1, 0] const nextX = x + dx[directionIndex] const nextY = y + dy[directionIndex]
根据当前的格子所在的位置x,y我们就知道接下来要走的位置,通过directionIndex的下标索引,知道我们下一个格子的坐标。
然后就是判断一下,边界的情况:
不能出界
判断能不能走
根据以上的信息,其实我们主要的逻辑部分就完成啦。
代码:
const spiralOrder = function (matrix) { if (matrix.length === 0) return []; const result = [], dx = [0, 1, 0, -1], dy = [1, 0, -1, 0], col = matrix.length, row = matrix[0].length; const isCheckMatrix = Array.from(new Array(col), () => (new Array(row).fill(false))) const dfs = (x, y, directionIndex) => { result.push(matrix[x][y]) // 存答案 isCheckMatrix[x][y] = true // 标记走过 for (let i = 0; i < 3; i++) { const nextX = x + dx[directionIndex] const nextY = y + dy[directionIndex] // 判断边界 if (nextX < col && nextX >= 0 && nextY < row && nextY >= 0 && !isCheckMatrix[nextX][nextY]) { dfs(nextX, nextY, directionIndex) } // 方向取余数 directionIndex = (directionIndex + 1) % 4; } }; dfs(0, 0, 0); return result };
这里我们需要对方向做余数处理。在确只有四个方向的情况,并且在这个方向不能走的情况下,尝试下一个方向。
directionIndex = (directionIndex + 1) % 4;
优化
写完的时候,我在想能不能优化一下,做个减枝的处理,后面发现,当前这个位置可以走的话,是不是就不能判断其他方向了。
或者说我们可以提前走出这个循环,这里做的优化就是return 当前的处理。
代码:
const spiralOrder = function (matrix) { if (matrix.length === 0) return []; const result = [], dx = [0, 1, 0, -1], dy = [1, 0, -1, 0], col = matrix.length, row = matrix[0].length; const isCheckMatrix = Array.from(new Array(col), () => (new Array(row).fill(false))) const dfs = (x, y, directionIndex) => { result.push(matrix[x][y]); isCheckMatrix[x][y] = true for (let i = 0; i < 3; i++) { const nextX = x + dx[directionIndex] const nextY = y + dy[directionIndex] if (nextX < col && nextX >= 0 && nextY < row && nextY >= 0 && !isCheckMatrix[nextX][nextY]) { return dfs(nextX, nextY, directionIndex) } directionIndex = (directionIndex + 1) % 4; } }; dfs(0, 0, 0); return result };
关于字节的笔试题和算法题示例分析就分享到这里了,希望以上内容可以对大家有一定的帮助,可以学到更多知识。如果觉得文章不错,可以把它分享出去让更多的人看到。