📑 题目：59. 螺旋矩阵 II

🚀 本题 LeetCode 传送门

题目大意

给定一个正整数 n，生成一个包含 1 到 n^2 所有元素，且元素按顺时针顺序螺旋排列的正方形矩阵。

解题思路

  • 给出一个数组 n，要求输出一个 n n 的二维数组，里面元素是 1 - nn，且数组排列顺序是螺旋排列的
  • 这一题是第 54 题的加强版，没有需要注意的特殊情况，直接模拟即可。

代码

  1. package leetcode
  3. func generateMatrix(n int) [][]int {
  4.  if n == 0 {
  5.   return [][]int{}
  6.  }
  7.  if n == 1 {
  8.   return [][]int{[]int{1}}
  9.  }
  10.  res, visit, round, x, y, spDir := make([][]int, n), make([][]int, n), 0, 0, 0, [][]int{
  11.   []int{0, 1},  // 朝右
  12.   []int{1, 0},  // 朝下
  13.   []int{0, -1}, // 朝左
  14.   []int{-1, 0}, // 朝上
  15.  }
  16.  for i := 0; i < n; i++ {
  17.   visit[i] = make([]int, n)
  18.   res[i] = make([]int, n)
  19.  }
  20.  visit[x][y] = 1
  21.  res[x][y] = 1
  22.  for i := 0; i < n*n; i++ {
  23.   x += spDir[round%4][0]
  24.   y += spDir[round%4][1]
  25.   if (x == 0 && y == n-1) || (x == n-1 && y == n-1) || (y == 0 && x == n-1) {
  26.    round++
  27.   }
  28.   if x > n-1 || y > n-1 || x < 0 || y < 0 {
  29.    return res
  30.   }
  31.   if visit[x][y] == 0 {
  32.    visit[x][y] = 1
  33.    res[x][y] = i + 2
  34.   }
  35.   switch round % 4 {
  36.   case 0:
  37.    if y+1 <= n-1 && visit[x][y+1] == 1 {
  38.     round++
  39.     continue
  40.    }
  41.   case 1:
  42.    if x+1 <= n-1 && visit[x+1][y] == 1 {
  43.     round++
  44.     continue
  45.    }
  46.   case 2:
  47.    if y-1 >= 0 && visit[x][y-1] == 1 {
  48.     round++
  49.     continue
  50.    }
  51.   case 3:
  52.    if x-1 >= 0 && visit[x-1][y] == 1 {
  53.     round++
  54.     continue
  55.    }
  56.   }
  57.  }
  58.  return res
  59. }