LeetCode-Spiral Matrix II

最新推荐文章于 2020-07-02 06:22:04 发布

BeHelium

最新推荐文章于 2020-07-02 06:22:04 发布

阅读量158

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签： LeetCode Java

本文链接：https://blog.youkuaiyun.com/qq_24133491/article/details/84579544

LeetCode 专栏收录该内容

306 篇文章

订阅专栏

本文介绍了一种螺旋矩阵生成算法，该算法接收一个整数n作为输入，返回一个n×n的方阵，矩阵中的元素从1到n²按螺旋顺序填充。通过使用状态数组和方向数组，算法能够有效地在螺旋路径上填充矩阵。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Description：
Given a positive integer n, generate a square matrix filled with elements from 1 to n2 in spiral order.

Example:

Input: 3
Output:
[
 [ 1, 2, 3 ],
 [ 8, 9, 4 ],
 [ 7, 6, 5 ]
]

题意：给定一个整数n，返回一个n阶方阵，其中的元素按照螺旋的方式填充元素1到n*n；如图所示：
在这里插入图片描述

解法：这一道题其实和Spiral Matrix是一样的，只不过一个是按照螺旋方式遍历元素，一个是按照螺旋方式填充元素，详细讲解可以参照Sprial Matrix的讲解；

Java

class Solution {
    public int[][] generateMatrix(int n) {
        int[][] result = new int[n][n];
        boolean[][] visited = new boolean[n][n];
        int cnt = 0;
        int[][] step = {{0, 1}, {1, 0}, {0, -1}, {-1, 0}};
        int stepIndex = 0;
        int row = 0;
        int col = 0;
        while (cnt++ < n * n) {
            result[row][col] = cnt;
            visited[row][col] = true;
            int tempRow = row + step[stepIndex][0];
            int tempCol = col + step[stepIndex][1];
            if (tempRow >= 0 && tempRow < n &&
            	tempCol >= 0 && tempCol < n &&
            	!visited[tempRow][tempCol]) {
            	row = tempRow;
            	col = tempCol;
            } else {
            	stepIndex = (stepIndex + 1) % step.length;
            	row += step[stepIndex][0];
            	col += step[stepIndex][1];
            }
        }
        return result;
    }
}