首页 > 代码库 > 【leetcode刷题笔记】Spiral Matrix II

【leetcode刷题笔记】Spiral Matrix II

2024-07-11 00:51:54   223人阅读

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

For example,
Given n = 3,

You should return the following matrix:

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

题解：以前做过的Spiral Matrix是给一个矩阵螺旋式的输出，这道题是给一个n，螺旋式的生成矩阵。和第一个的做法类似，定义四个变量

row_left:当前未填的子矩阵的最左边界

row_right:当前未填的子矩阵的最右边界

column_up:当前未填子矩阵的最上边界

column_down:当前未填子矩阵的最下边界

然后每次进行四个循环，做一个螺旋。

最后一个元素要单独处理（其实只要n是奇数的时候单独处理），因为这时只有一个元素未填，不够运行一个螺旋，如果还是用四个for循环就会重复修改已经放置好的元素。

代码如下：

 1 public class Solution { 2     public int[][] generateMatrix(int n) { 3         int matrix[][] = new int[n][n]; 4         int row_up = 0; 5         int row_down = n - 1; 6         int column_left = 0; 7         int column_right = n - 1; 8         int number = 1; 9         10         while(number <= n*n-1){11             for(int i = column_left;i <= column_right;i++){12                 matrix[row_up][i] = number;13                 number++;14             }15             row_up++;16             17             for(int j = row_up;j <= row_down;j++){18                 matrix[j][column_right] = number;19                 number++;20             }21             column_right--;22             23             for(int j = column_right;j >= column_left;j --){24                 matrix[row_down][j] = number;25                 number++;26             }27             row_down--;28             29             for(int i = row_down;i >= row_up;i --){30                 matrix[i][column_left] = number;31                 number++;32             }33             column_left++;34             35         }36         37         if(n > 0)38             matrix[n/2][n%2==0?n/2-1:n/2] = n*n;39         return matrix;40     }41 }

还有个坑就是n等于0的时候，就没有所谓的“最后一个元素单独处理了”，这中情况要判断，否则会抛出异常。

 


联系
我们