螺旋矩阵 II(Java)

(leetcode.59)

题目要求

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

思路

1.向上赋值和向下赋值为一组，向左赋值和向右赋值为一组
2.每次为二维数组某一行或某一列赋值的次数有如下规律

n=5        左5下4  右4上3  左3下2  右2上1  左1右1
              5 4 4 3 3 2 2 1 1
n=6        6 5 5 4 4 3 3 2 2 1 1

注意

1.边界条件和防止数组越界是最重要的，中间一直数组越界报错，我花了好久才把边界条件修改明白😢😢😢
2.判断条件 (j + 1 < matrix[0].length && matrix[i][j + 1] == 0)
这里为了防止matrix[i][j + 1]数组越界报错，将其放在&&后面，这样当前面条件不成立，后面直接跳过
3.每次自增和自减放在前面也是为了防止数组越界，要是放在后面，则需要在for循环后再加一个自增或自减，更麻烦

代码

class Solution {
    public int[][] generateMatrix(int n) {
       int flag = 1;
       int i = 0;
       int j = -1;
       int[][] matrix = new int[n][n];

       while(n > 0) {
           if(j + 1 < matrix[0].length && matrix[i][j + 1] == 0) {//左移
               for(int x = 0; x < n; x ++) {
                   j ++;
                   matrix[i][j] = flag;
                   flag ++;
               }
           } else {//右移
               for(int x = 0; x < n; x ++) {
                   j --;
                   matrix[i][j] = flag;
                   flag ++;
               }
           }
           n --;
           if(i + 1 < matrix[0].length && matrix[i + 1][j] == 0) {//下移
               for(int x = 0; x < n; x ++) {
                   i ++;
                   matrix[i][j] = flag;
                   flag ++;
               }
           } else {//上移
               for(int x = 0; x < n; x ++) {
                   i --;
                   matrix[i][j] = flag;
                   flag ++;
               }
           }
       }
       return matrix;
    }
}