修改 0059 螺旋矩阵II JavaScript版本 解法

提供一种条理更清晰的方法,与本文介绍的解法更吻合
This commit is contained in:
Jerry-306
2021-09-24 16:45:41 +08:00
committed by GitHub
parent 0f0bb6e741
commit b4e6579793

View File

@ -239,45 +239,54 @@ class Solution:
javaScript javaScript
```js ```javascript
/** /**
* @param {number} n * @param {number} n
* @return {number[][]} * @return {number[][]}
*/ */
var generateMatrix = function(n) { var generateMatrix = function(n) {
// new Array(n).fill(new Array(n)) let startX = startY = 0; // 起始位置
// 使用fill --> 填充的是同一个数组地址 let loop = Math.floor(n/2); // 旋转圈数
const res = Array.from({length: n}).map(() => new Array(n)); let mid = Math.floor(n/2); // 中间位置
let loop = n >> 1, i = 0, //循环次 let offset = 1; // 控制每一层填充元素个
count = 1, let count = 1; // 更新填充数字
startX = startY = 0; // 起始位置 let res = new Array(n).fill(0).map(() => new Array(n).fill(0));
while(++i <= loop) {
// 定义行列 while (loop--) {
let row = startX, column = startY; let row = startX, col = startY;
// [ startY, n - i) // 上行从左到右(左闭右开)
while(column < n - i) { for (; col < startY + n - offset; col++) {
res[row][column++] = count++; res[row][col] = count++;
} }
// [ startX, n - i) // 右列从上到下(左闭右开)
while(row < n - i) { for (; row < startX + n - offset; row++) {
res[row++][column] = count++; res[row][col] = count++;
} }
// [n - i , startY) // 下行从右到左(左闭右开)
while(column > startY) { for (; col > startX; col--) {
res[row][column--] = count++; res[row][col] = count++;
} }
// [n - i , startX) // 左列做下到上(左闭右开)
while(row > startX) { for (; row > startY; row--) {
res[row--][column] = count++; res[row][col] = count++;
} }
startX = ++startY;
// 更新起始位置
startX++;
startY++;
// 更新offset
offset += 2;
} }
if(n & 1) { // 如果n为奇数的话需要单独给矩阵最中间的位置赋值
res[startX][startY] = count; if (n % 2 === 1) {
res[mid][mid] = count;
} }
return res; return res;
}; };
``` ```
Go: Go: