leetcode 59. Spiral Matrix II（螺旋矩阵II）

关注

原创 2023-05-10 09:02:36 351 阅读 AI 写同款· GEO 优化 ›

Python输出螺旋形数组 Python输出螺旋形数组思路思路 1、先初始化一个二维数组，从外向内，一圈圈填充。 2、每一圈由每组3个，共4组数构成。确定起点、数量、和方向即可用4个循环完成填充。 3、简化思路：循环第1组（蓝色），相应可以由步长3确定后面3组的位置，即循环1次即可。按上图标数顺序依次填充即可完成一圈。代码如下： #!/usr/bin/python3 # -*- coding: utf-8 _*_ ... 阅读详情

Given a positive integer n, generate a square matrix filled with elements from 1 to n2 in spiral order.

Example:

Input: 3
Output:
[
[ 1, 2, 3 ],
[ 8, 9, 4 ],
[ 7, 6, 5 ]
]

给出一个数字n，让输出一个二维数组，元素是从1～n²，而且要是螺旋方式

思路：
从1～n²很简单，主要是螺旋方式，螺旋方式可以参考54题，还是两条边两条边地输出，先是第一行和最后一列，然后
迂回最后一行和第一列，如此循环。

注意每个while (row <= endRow && col <= endCol) {结束时，row都是多一行或者少一行的，
而且为了避免位置重复，列也是要+1或者-1

//0ms
    public int[][] generateMatrix(int n) {
        int[][] result = new int[n][n];
        
        int element = 1;
        int lastE = n * n;
        
        int startRow = 0;
        int endRow = n - 1;
        int startCol = 0;
        int endCol = n - 1;
        
        
        int row = startRow;
        int col = startCol;
        
        while (element <= lastE) {
            while (row <= endRow && col <= endCol) {
                result[row][col] = element;
                if (col == endCol) {
                    row ++;
                } else {
                    col ++;
                }
                element ++;
            }
            row --;
            col --;
            startRow ++;
            endCol --;
            
            while (row >= startRow && col >= startCol) {
                result[row][col] = element;
                if (col == startCol) {
                    row --;
                } else {
                    col --;
                }
                element ++;
            }
            row ++;
            col ++;
            endRow --;
            startCol ++;
            
        }
        return result;
    }

另一版本

    public int[][] generateMatrix(int n) {
        int[][] res = new int[n][n];
        int sR = 0;
        int eR = n - 1;
        int sC = 0;
        int eC = n - 1;
        int row = 0;
        int col = 0;
        int i = 1;
            
        while(sR <= eR) {         
            if(row == sR) {
               while(col <= eC) {
                   res[row][col] = i ++;
                   col ++;
               }
               col = eC;
               row ++;
               while(row <= eR) {
                   res[row][col] = i ++;
                   row ++;
               }
                row = eR;
                col --;
                sR ++;
                eC --;
            }
            
            if(row == eR) {
                while(col >= sC) {
                    res[row][col] = i ++;
                    col --;
                }
                col = sC;
                row --;
                while(row >= sR) {
                    res[row][col] = i ++;
                    row --;
                }
                row =  sR;
                col ++;
                eR --;
                sC ++;
            }
        }
        return res;
    }

public int[][] generateMatrix(int n) {
    int[][] res = new int[n][n];
    int cnt = 1;

    int minR = 0;
    int maxR = n-1;
    int minC = 0;
    int maxC = n-1;
    int r = 0;
    int c = 0;

    while(minR <= maxR && minC <= maxC) {
        if(minR <= maxR && minC <= maxC) {
            while(c <= maxC) {
                res[r][c] = cnt++;
                c ++;
            }
            c = maxC;
            minR ++;
            r = minR;
            while(r <= maxR){
                res[r][c] = cnt++;
                r++;
            }
            r = maxR;
            maxC--;
            c = maxC;
        }
        if(minR <= maxR && minC <= maxC) {
            while(c >= minC) {
                res[r][c] = cnt++;
                c --;
            }
            c = minC;
            maxR --;
            r = maxR;
            while(r >= minR){
                res[r][c] = cnt++;
                r--;
            }
            r = minR;
            minC++;
            c = minC;
        }
    }
    return res;
}