Leetcode每日一题-59.螺旋矩阵Ⅱ

59. 螺旋矩阵Ⅱ

题目链接

描述

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

示例

示例1：

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

示例2：

输入：n = 1
输出：[[1]]

提示

• 1 <= n <= 20

分析

1. 首先，看提示：n的范围给定了1到20，所以不必在进行有效性的判定
2. 此题和第54题是类似题，都是蛇形矩阵相关题
3. 区别在于54题是取值，本题是放值，直接将54的代码进行修改就能满足需求

代码

func generateMatrix(n int) [][]int {
    
    
	// 初始化
	ret := make([][]int, n)
	for i := 0; i < n; i++ {
    
    
		ret[i] = make([]int, n)
	}
	// flag用来判断转方向后是否发生了移动，如果连续转了两次方向，那么说明遍历结束了
	flag := true
	// 定义方向
	// right = {0,1},down = {1,0}, left = {0,-1}, up = {-1, 0}
	// directs 是上述的方向集
	directs := [][]int{
    
    {
    
    0, 1}, {
    
    1, 0}, {
    
    0, -1}, {
    
    -1, 0}}
    // aim表示要放的目标个数
	aim := n * n
	// 初始向右，定义方向索引p 为0
	p := 0
	direct := directs[p]
	// x， y为遍历matrix的行列指针
	x := 0
	y := 0
	ret[0][0] = 1
	for i := 1; i < aim; {
    
    
		// m, n移动
		x += direct[0]
		y += direct[1]
		// 如果当前值为101，说明遍历过了，==row,==col 说明到达边界了，转方向
		if x == n || y == n || x < 0 || y < 0 || ret[x][y] != 0 {
    
    
			// 当前位置不对，退一步
			x -= direct[0]
			y -= direct[1]
			// 退一步之后判断flag，为true那么转方向继续
			if flag {
    
    
				p++
				p = p % 4
				direct = directs[p]
				flag = false
				continue
			}
			// 如果flag 为false，那么就是连续发生了两次转向，未移动，那么说明遍历结束了
			return ret
		}
		ret[x][y] = i+1
		i++
		// flag = true，表示有移动
		flag = true
	}
	return ret
}

提交结果

蛇形矩阵可以使用不断缩小循环范围的方式去遍历，不设置方向，这样可以在空间上也得到节省。