leetcode59. 螺旋矩阵 II

最新推荐文章于 2023-03-31 11:00:00 发布

白哥学算法

最新推荐文章于 2023-03-31 11:00:00 发布

阅读量202

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/qq_39448574/article/details/89600120

版权

算法专栏收录该内容

105 篇文章 0 订阅

订阅专栏

题目

给定一个正整数 n，生成一个包含 1 到 n2 所有元素，且元素按顺时针顺序螺旋排列的正方形矩阵。
示例:
输入: 3
输出: [
[ 1, 2, 3 ],
[ 8, 9, 4 ],
[ 7, 6, 5 ] ]

解题思路

从矩形的外部一直往内部螺旋的加入数字就好了，类似于之前的螺旋矩阵1的解法，但是这次是正方形，所以这题的下标更好理解。

代码如下

class Solution {
    public int[][] generateMatrix(int n) {
        int[][] res = new int[n][n];
        //代表数组长度res.length-1;
        n--;
        int num = 1;
        for(int i = 0; i <= n/2; i ++) {
        	//剩下最后一个的情况
        	if(n == 2*i) {
        		res[i][i] = num;
        		break;
        	};
        	num = addRow(res, i, i, n-i, num);
        	num = addCol(res, n-i, i, n-i, num);
        	num = addRow(res, n-i, n-i, i, num);
        	num = addCol(res, i, n-i,i, num);
        }
        return res;
    }
	//返回值保存num的值
	public int addRow(int[][] res,int row, int colStart, int colEnd, int num) {
		if(colStart == colEnd) {
			res[row][colStart] = num++;
			return num;
		}
		for(int i = colStart;i < colEnd; i ++) res[row][i] = num++;
		for(int i = colStart;i > colEnd; i --) res[row][i] = num++;
		return num;
	}
	public int addCol(int[][] res,int col, int rowStart, int rowEnd, int num) {
		if(rowStart == rowEnd) {
			res[col][rowEnd] = num++;
			return num;
		}
		for(int i = rowStart;i < rowEnd; i ++) res[i][col] = num++;
		for(int i = rowStart;i > rowEnd; i --) res[i][col] = num++;
		return num;
	}
}