【数组】leetcode_59_螺旋矩阵II

【数组】leetcode_59_螺旋矩阵II

c++

指针法：上下左右4个指针调整赋值

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> res(n, vector<int>(n, 0));
        int left = 0;
        int right = n-1;
        int top = 0;
        int bottom = n-1;
        int count = 1;

        while(count <= n*n){
            for(int i = left; i <= right; i++){
                res[top][i] = count;
                count++;
            }
            top++;

            for(int i = top; i <= bottom; i++){
                res[i][right] = count;
                count++;
            }
            right--;

            for(int i = right; i >= left; i-- ){
                res[bottom][i] = count;
                count++;
            }
            bottom--;

            for (int i = bottom; i>= top;i--){
                res[i][left] = count;
                count++;
            }
            left++;
        }
        return res;
    }
};

偏移量法

大佬讲解-注意循环不变量

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> res(n, vector<int>(n, 0)); // 使用vector定义一个二维数组
        int startx = 0, starty = 0; // 定义每循环一个圈的起始位置
        int loop = n / 2; // 每个圈循环几次，例如n为奇数3，那么loop = 1 只是循环一圈，矩阵中间的值需要单独处理
        int mid = n / 2; // 矩阵中间的位置，例如：n为3， 中间的位置就是(1，1)，n为5，中间位置为(2, 2)
        int count = 1; // 用来给矩阵中每一个空格赋值
        int offset = 1; // 每一圈循环，需要控制每一条边遍历的长度
        int i,j;
        while (loop --) {
            i = startx;
            j = starty;

            // 下面开始的四个for就是模拟转了一圈
            // 模拟填充上行从左到右(左闭右开)
            for (j = starty; j < starty + n - offset; j++) {
                res[startx][j] = count++;
            }
            // 模拟填充右列从上到下(左闭右开)
            for (i = startx; i < startx + n - offset; i++) {
                res[i][j] = count++;
            }
            // 模拟填充下行从右到左(左闭右开)
            for (; j > starty; j--) {
                res[i][j] = count++;
            }
            // 模拟填充左列从下到上(左闭右开)
            for (; i > startx; i--) {
                res[i][j] = count++;
            }

            // 第二圈开始的时候，起始位置要各自加1， 例如：第一圈起始位置是(0, 0)，第二圈起始位置是(1, 1)
            startx++;
            starty++;

            // offset 控制每一圈里每一条边遍历的长度
            offset += 2;
        }

        // 如果n为奇数的话，需要单独给矩阵最中间的位置赋值
        if (n % 2) {
            res[mid][mid] = count;
        }
        return res;
    }
};

python

res = n * [n * [0]] 这里构造二维数组的时候发现了个问题
详细见这里

class Solution:
    def generateMatrix(self, n):
        # res = n * [n * [0]]  # 这里发现了个问题
        res = [[0 for i in range(n)] for j in range(n)]
        startx = 0
        starty = 0
        loop = n // 2
        count = 1
        offset = 1
        mid = n // 2

        while (count < n * n):
            i = startx
            j = starty

            while (j < starty + n - offset):
                res[i][j] = count
                count += 1
                j += 1

            while (i < startx + n - offset):
                res[i][j] = count
                count += 1
                i += 1

            while (j > starty):
                res[i][j] = count
                count += 1
                j -= 1

            while (i > startx):
                res[i][j] = count
                count += 1
                i -= 1

            startx += 1
            starty += 1
            offset += 2

        if n % 2 != 0:
            res[mid][mid] = count

        return res