美文网首页
leetcode--59--螺旋矩阵 II

leetcode--59--螺旋矩阵 II

作者: minningl | 来源:发表于2021-06-27 18:35 被阅读0次

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

    示例 1:


    image.png

    输入:n = 3
    输出:[[1,2,3],[8,9,4],[7,6,5]]
    示例 2:

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

    提示:

    1 <= n <= 20

    链接:https://leetcode-cn.com/problems/spiral-matrix-ii

    思路:
    1、模拟法,按照题目要求的顺序进行遍历

    Python代码:

    class Solution(object):
        def generateMatrix(self, n):
            """
            :type n: int
            :rtype: List[List[int]]
            """
            arr = [[0 for i in range(n)] for j in range(n)]
    
            cnt = 1
            j = 0
            while cnt<=n*n:
                for i in range(j, n-j):
                    arr[j][i] = cnt
                    cnt += 1
                for i in range(j+1, n-j):
                    arr[i][n-j-1] = cnt
                    cnt += 1
                for i in range(n-j-2, j-1, -1):
                    arr[n-j-1][i] = cnt
                    cnt += 1
                for i in range(n-j-2, j, -1):
                    arr[i][j] = cnt
                    cnt += 1
                j += 1
            return arr
    

    Java代码:

    class Solution {
        public int[][] generateMatrix(int n) {
            int [][] arr = new int[n][n];
            int cnt = 1;
            int j = 0;
    
            while(cnt<=n*n){
                for(int i=j; i<n-j; i++){
                    arr[j][i] = cnt++;
                }
                for(int i=j+1; i<n-j; i++){
                    arr[i][n-j-1] = cnt++;
                }
                for(int i=n-j-2; i>=j; i--){
                    arr[n-j-1][i] = cnt++;
                }
                for(int i=n-j-2; i>j; i--){
                    arr[i][j] = cnt++;
                }
                j += 1;
            }
    
            return arr;
        }
    }
    

    相关文章

      网友评论

          本文标题:leetcode--59--螺旋矩阵 II

          本文链接:https://www.haomeiwen.com/subject/yzzuyltx.html