螺旋矩阵II

螺旋矩阵II

59 螺旋矩阵II：

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

示例 1：

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

示例 2：

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

如果要写出正确的二分法一定要坚持循环不变量原则。

而求解本题依然是要坚持循环不变量原则。

模拟顺时针画矩阵的过程:

• 填充上行从左到右
• 填充右列从上到下
• 填充下行从右到左
• 填充左列从下到上

由外向内一圈一圈这么画下去。

可以发现这里的边界条件非常多, 这里一圈下来，要画每四条边，这四条边怎么画，每画一条边都要坚持一致的左闭右开，或者左开右闭的原则，这样这一圈才能按照统一的规则画下来。

按照左闭右开的原则，来画一圈：

这里每一种颜色，代表一条边，我们遍历的长度，可以看出每一个拐角处的处理规则，拐角处让给新的一条边来继续画。

这也是坚持了每条边左闭右开的原则。

左闭右开原则：时间复杂度 O(n^2)

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        // 使用vector定义一个二维数组
        vector<vector<int>> arr(n, vector<int>(n, 0));
        int startx = 0, starty = 0; //每一圈开始的起始位置
        int loop = n / 2;// 每个圈循环几次，例如n为奇数3，那么loop = 1 只是循环一圈，矩阵中间的值需要单独处理
        int mid = n / 2;// 矩阵中间的位置
        int count = 1; // 用来给矩阵中每一个空格赋值
        int offset = 1; // 需要控制每一条边遍历的长度，每次循环右边界收缩一位
        int i,j;
        while(loop--) {
            i = startx;
            j = starty;
            // 下面开始的四个for就是模拟转了一圈
            // 模拟填充上行从左到右(左闭右开)
            for(j = starty; j < n - offset; j++){
                arr[startx][j] = count++;
            }
            // 模拟填充右列从上到下(左闭右开)
            for(i = startx; i < n - offset; i++){
                arr[i][j] = count++;
            }
            // 模拟填充下行从右到左(左闭右开)
            for(;j > starty; j--) {
                arr[i][j] = count++;
            }
             // 模拟填充左列从下到上(左闭右开)
             for(; i > startx; i--) {
                 arr[i][j] = count++;
             }
             // 第二圈开始的时候，起始位置要各自加1。
            startx++;
            starty++;
            // offset 控制每一圈里每一条边遍历的长度
            offset += 1;
        }
         // 如果n为奇数的话，需要单独给矩阵最中间的位置赋值
        if (n % 2) {
            arr[mid][mid] = count;
        }
        return arr;

    }
};

相关题目

54 螺旋矩阵

剑指offer29 顺时针打印矩阵

给你一个 m 行 n 列的矩阵 matrix ，请按照 顺时针螺旋顺序 ，返回矩阵中的所有元素。

示例 1：

输入：matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出：[1,2,3,6,9,8,7,4,5]

示例 2：

输入：matrix = [[1,2,3,4],[5,6,7,8],[9,10,11,12]]
输出：[1,2,3,4,8,12,11,10,9,5,6,7]

提示：

m == matrix.length
n == matrix[i].length
1 <= m, n <= 10
-100 <= matrix[i][j] <= 100

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        vector<int> res;
        if(matrix.empty()){
            return res;
        }
        int up = 0, down = matrix.size() - 1 ;  //上下界
        int left = 0, right = matrix[0].size() - 1; // 左右界

        while(true) {
            //左->右
            for(int i = left; i <= right; ++i){
                res.push_back(matrix[up][i]);
            }
            if(++up > down) {
                break;
            }
            //上->下
            for(int i = up; i <= down; ++i){
                res.push_back(matrix[i][right]);
            }
            if(--right < left) {
                break;
            }
            //右->左
            for(int i = right; i >= left; --i){
                res.push_back(matrix[down][i]);
            }
            if(--down < up) {
                break;
            }
            //下->上
            for(int i = down; i >= up; --i){
                res.push_back(matrix[i][left]);
            }
            if(++left > right) {
                break;
            }

        }
        return res;

    }
};