54. 螺旋矩阵

54. 螺旋矩阵

代码

javascript

// 54. 螺旋矩阵：https://leetcode.cn/problems/spiral-matrix/description/
// 输入：matrix = [[1,2,3],[4,5,6],[7,8,9]]
// 输出：[1,2,3,6,9,8,7,4,5]

export function spiralMatrix (matrix) {
  if (!matrix.length || !matrix[0].length) {
    return [];
  }
  let top = 0
  let left = 0
  let bottom = matrix.length - 1
  let right = matrix[0].length - 1
  const res = []
  while (top <= bottom && left <= right) {
    for (let col = left; col <= right; col++) {
      res.push(matrix[top][col])
    }
    for (let row = top + 1; row <= bottom; row++) {
      res.push(matrix[row][right])
    }
    if (top < bottom && left < right) {
      for (let col = right - 1; col >= left; col--) {
        res.push(matrix[bottom][col])
      }
      for (let row = bottom - 1; row > top; row--) {
        res.push(matrix[row][left])
      }
    }
    top += 1
    left += 1
    bottom -= 1
    right -= 1
  }
  return res
}

typescript

// 54. 螺旋矩阵：https://leetcode.cn/problems/spiral-matrix/description/
// 输入：matrix = [[1,2,3],[4,5,6],[7,8,9]]
// 输出：[1,2,3,6,9,8,7,4,5]

export function spiralMatrix (matrix: number[][]): number[] {
  if (!matrix.length || !matrix[0].length) {
    return [];
  }
  let top = 0
  let left = 0
  let bottom = matrix.length - 1
  let right = matrix[0].length - 1
  const res: number[] = []
  while (top <= bottom && left <= right) {
    for (let col = left; col <= right; col++) {
      res.push(matrix[top][col])
    }
    for (let row = top + 1; row <= bottom; row++) {
      res.push(matrix[row][right])
    }
    if (top < bottom && left < right) {
      for (let col = right - 1; col >= left; col--) {
        res.push(matrix[bottom][col])
      }
      for (let row = bottom - 1; row > top; row--) {
        res.push(matrix[row][left])
      }
    }
    top += 1
    left += 1
    bottom -= 1
    right -= 1
  }
  return res
}

测试代码

test

import { expect, test } from 'vitest'
import { spiralMatrix } from './typescript.ts'
import { spiralMatrix as spiralMatrixJs } from './javascript.js'

test(`spiralMatrix`, () => {
  expect(spiralMatrix([[1,2,3,4],[5,6,7,8],[9,10,11,12]])).toEqual([1,2,3,4,8,12,11,10,9,5,6,7])
  expect(spiralMatrix([])).toEqual([])
})

test(`spiralMatrixJs`, () => {
  expect(spiralMatrixJs([[1,2,3],[4,5,6],[7,8,9]])).toEqual([1,2,3,6,9,8,7,4,5])
  expect(spiralMatrixJs([[]])).toEqual([])
})