Leetcode的AC指南 —— 数组：59.螺旋矩阵II —— 循环不变量原则

摘要：
Leetcode的AC指南 —— 数组：59.螺旋矩阵II —— 循环不变量原则。题目介绍：给你一个正整数 n ，生成一个包含 1 到 n2 所有元素，且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。

一、题目

---

题目介绍：给你一个正整数 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、循环不变量原则

求解本题是要坚持循环不变量原则。

模拟顺时针画矩阵的过程:

• 填充上行从左到右
• 填充右列从上到下
• 填充下行从右到左
• 填充左列从下到上
  由外向内一圈一圈这么画下去

这里一圈下来，我们要画每四条边，这四条边怎么画，每画一条边都要坚持一致的左闭右开，或者左开右闭的原则，这样这一圈才能按照统一的规则画下来。

那么我按照左闭右开的原则，来画一圈，大家看一下：

这里每一种颜色，代表一条边，我们遍历的长度，可以看出每一个拐角处的处理规则，拐角处让给新的一条边来继续画。

这也是坚持了每条边左闭右开的原则。

一些同学做这道题目之所以一直写不好，代码越写越乱。

就是因为在画每一条边的时候，一会左开右闭，一会左闭右闭，一会又来左闭右开，岂能不乱。

代码如下，已经详细注释了每一步的目的，可以看出while循环里判断的情况是很多的，代码里处理的原则也是统一的左闭右开。

class Solution {
    public int[][] generateMatrix(int n) {
        int loop = 0;  // 控制循环次数
        int[][] res = new int[n][n];
        int start = 0;  // 每次循环的开始点(start, start)
        int count = 1;  // 定义填充数字
        int i, j;

        while (loop++ < n / 2) { // 判断边界后，loop从1开始
            // 模拟上侧从左到右
            for (j = start; j < n - loop; j++) {
                res[start][j] = count++;
            }

            // 模拟右侧从上到下
            for (i = start; i < n - loop; i++) {
                res[i][j] = count++;
            }

            // 模拟下侧从右到左
            for (; j >= loop; j--) {
                res[i][j] = count++;
            }

            // 模拟左侧从下到上
            for (; i >= loop; i--) {
                res[i][j] = count++;
            }
            start++;
        }

        if (n % 2 == 1) {
            res[start][start] = count;
        }

        return res;
    }
}

三、总结

---

我的代码

public int[][] generateMatrix(int n) {

        int len = n;
        int[][] nums = new int[n][n];
        int num = 1;
        int x,y;

        for(int i = 0; i < n; i++){
        	//缩圈
            x = i;
            y = i;
            
            //从左到右
            for(x = i; x < n; x++) nums[i][x] = num++;
            
			//从上到下
            for(y = i + 1; y < n; y++) nums[y][n - 1] = num++;
			//从右到左
            for(x = n - 2; x >= i; x--) nums[n - 1][x] = num++;
            //从下到上
            for(y = n - 2; y > i; y--) nums[y][i] = num++;
			
            n--;
        }
        for(int i = 0; i < len; i++)
            System.out.println(Arrays.toString(nums[i]));

        return nums;

    }

优化后的代码：

	public int[][] generateMatrix(int n) {
		
        int len = n - 1; //每圈左闭右开的长度
        int[][] nums = new int[n][n];
        int num = 1;
        int start = 0; //每圈的起始点
        int x;
        int y;
        while(len > 0){
            x = start;
            y = start;
            //从左到右
            for(; x < len ; x++) nums[start][x] = num++;
            //从上到下
            for(; y < len ;  y++) nums[y][len] = num++;
            //从右到左
            for(; x > start; x--) nums[len][x] = num++;
            从下到上
            for(; y > start; y--) nums[y][start] = num++;

            start++; //更新起始点
            len--; // 更新每圈左闭右开的长度
        }
		
        if(n % 2 == 1) nums[n / 2][n / 2] = num; // 为数组中心数据赋值

        for(int i = 0; i < n; i++)
            System.out.println(Arrays.toString(nums[i]));

        return nums;

    }