「日更刷题」59.螺旋矩阵

238 阅读1分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第1天,点击查看活动详情

一、题目描述:

59.螺旋矩阵-力扣

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

实例1:

输入: n = 3
输出: [[1,2,3],[8,9,4],[7,6,5]]
解释: n = 3 代表是由数字19组成了一个 int[3][3] 的数组

image.png

实例2:

输入: n = 1
输出: [[1]]

提示: 1 <= n <= 20

二、思路分析

模拟填入矩阵的过程:

  • 从上行左到右
  • 从右侧上到下
  • 从下行右到左
  • 从左侧下到上

使用二分法判断填入矩阵的循环次数, 三次有两圈, 六次有三圈, 九次有五圈. 这里要注意的是最后要加个判断, 如果是单数最中间的数字要加进去

采用左闭右开顺序添加数字, 避免后期混乱容易出错, 这里借用卡尔的一张图

image.png

三、java代码

public int[][] generateMatrix(int n) {
       int loopNum = 0;
        int[][] res = new int[n][n];
        int start = 0;
        int count = 1;
        int i,j;

        while(loopNum++ < n /2){
            for (i = start; i< n - loopNum;i++){
                res[start][i] = count;
                count++;
            }
            for (j = start ; j < n-loopNum; j++){
                res[j][i] = count;
                count++;
            }
            for (; i >= loopNum;i--){
                res[j][i] = count;
                count++;
            }
            for (; j >= loopNum; j--){
                res[j][start] = count;
                count++;
            }
            start ++;
        }
        if (n % 2 == 1){
            res[start][start] = count;
        }
        return res;
    }

四、总结

最开始看着还是比较难的, 但是真正的实践起来之后就也还好, 中间碰到一点小问题都是边界没有找好.

五、参考

代码随想录 - 螺旋矩阵思路