LeetCode:螺旋矩阵


题目

给你一个 mn 列的矩阵 matrix ,请按照 顺时针螺旋顺序 ,返回矩阵中的所有元素。

示例 1:

1
2
输入:matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出:[1,2,3,6,9,8,7,4,5]

示例 2:

1
2
输入:matrix = [[1,2,3,4],[5,6,7,8],[9,10,11,12]]
输出:[1,2,3,4,8,12,11,10,9,5,6,7]

提示:

  • m == matrix.length
  • n == matrix[i].length
  • 1 <= m, n <= 10
  • -100 <= matrix[i][j] <= 100

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/spiral-matrix

思路

如图顺序:

  • 第一行正常顺序
  • 非首尾行取最后一个元素
  • 尾行倒序
  • 在非首尾行的首列倒序
  • 剩下黄色区域复用以上逻辑(递归)

代码

Go

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
package main

import "fmt"

func spiralOrder(matrix [][]int) []int {
var result []int
var cl []int // 非首尾行的首字母集合
var cc [][]int // 非首尾行中间字母多维集合

for i, m := range matrix {
matrix_len := len(matrix)

if i == 0 {
result = append(result, m...)
}

if i > 0 && i < matrix_len-1 {
m_len := len(m)

result = append(result, m[m_len-1])

if matrix_len >= 3 && m_len >= 2 {
cl = append([]int{m[0]}, cl...)
}

if m_len >= 3 {

cc = append(cc, m[1:m_len-1])
}
}

if matrix_len > 1 && i == matrix_len-1 {
result = append(result, reverseInts(m)...)
}
}

result = append(result, cl...)
if len(cc) > 0 {
result = append(result, spiralOrder(cc)...)
}

return result
}

// 反转
func reverseInts(m []int) []int {
var r []int

for i := range m {
r = append(r, m[len(m)-1-i])
}

return r
}

func main() {
fmt.Println(reverseInts([]int{1, 2, 3}))

fmt.Println(spiralOrder([][]int{
{1, 2, 3}, {4, 5, 6}, {7, 8, 9},
}))

fmt.Println(spiralOrder([][]int{
{2, 5}, {8, 4}, {0, -1},
}))
}

评论