代码随想录数组篇 —— 螺旋矩阵II

题目

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

思路

注意循环不变量原则即可，不要一会左闭右开，一会左闭右闭，要始终保持相同的边界条件画下去。

然后按照上行从左到右—>右列从上到下—>下行从右到左—>左列从下到上的顺序循环走下去。

代码（java版本）

//左闭右开
class Solution {
    public int[][] generateMatrix(int n) {
        int[][] res = new int[n][n];
        int startx = 0,starty = 0;//定义每个循环的圈的起始位置
        int loop = n/2;//圈的循环次数
        int mid = n/2;//矩阵最中间的位置，单独填充
        int count = 1;
        int offset = 1;//控制每次循环的圈的遍历长度，每次循环右边界减小一位
        int i,j;
        while(loop -- > 0){
            i = startx;
            j = starty;
			//上行从左到右
            for(j = starty;j < n - offset;j++){//相当于第一行的 0 ~ n-2 个格子，n-1个格子走不到，因为是右开，留给了后面
                res[startx][j] = count++;
            }
			//右列从上到下
            for(i = startx;i < n - offset;i++){//相当于最后一列的 0 ~ n-2 个格子
                res[i][j] = count++;
            }
			//下行从右到左
            for (; j > starty ; j--) {
                res[i][j] = count++;
            }
			//左列从下到上
            for(;i > startx;i--){
                res[i][j] = count++;
            }
			//第二圈开始的时候，起始位置要各自加1， 例如：第一圈起始位置是(0, 0)，第二圈起始位置是(1, 1)
            startx++;
            starty++;
            //offset 控制每一圈里每一条边遍历的长度
            offset++;
        }
         如果n为奇数的话，需要单独给矩阵最中间的位置赋值
        if(n%2 == 1){
            res[mid][mid] = count;
        }
        return  res;
    }
}