Leetcode 二维数组的自定义遍历

介绍

二维数组就是在一维数组上多加一个维度，在C++中，也可以理解为在一维数组的基础上，每个元素都是一维数组，所以它的地址是连续的。

题目

54. 螺旋矩阵

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

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        vector<int> ans = {};
        int i = -1 , j = -1, loop = min(matrix.size()/2, matrix[0].size()/2), start = loop;
        while (loop){
            for (i = start - loop, j = start - loop; j < matrix[0].size()-1 - start + loop; j++) {
                ans.push_back(matrix[i][j]);        
            }
            
            for (; i < matrix.size()-1 - start + loop; i++) {
                ans.push_back(matrix[i][j]);
            }
            
            for (; j > start - loop; j--){
                ans.push_back(matrix[i][j]);
            }
            
            for (; i > start - loop; i--){
                ans.push_back(matrix[i][j]);
            }
            loop--;
        }
        i++;
        j++;
        //有单独成行或成列的序列块
        if (i == j){
            for (int s = i; s < matrix.size() - i; s++) {
                for (int t = j; t < matrix[0].size() - j; t++) {
                    ans.push_back(matrix[s][t]);
                }
            }
        }
        return ans;
    }
};

59. 螺旋矩阵 II

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

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> ans;
        ans.resize(n);
        for (int i = 0; i < n; i++) {
            ans[i].resize(n);
        }
        int i = -1 , j = -1, loop = n/2, start = loop, token = 1;
        while (loop){
            for (i = start - loop, j = start - loop; j < n -1 - start + loop; j++) {
                ans[i][j] = token;
                token++;        
            }
            
            for (; i < n -1 - start + loop; i++) {
                ans[i][j] = token;
                token++;
            }
            
            for (; j > start - loop; j--){
                ans[i][j] = token;
                token++;
            }
            
            for (; i > start - loop; i--){
                ans[i][j] = token;
                token++;
            }
            loop--;
        }
        i++;
        j++;
        if (i == j && i < n - i){
            ans[i][j] = token;
        }
        return ans;
    }
};

值得注意的是，vector.resize()带两个参数，一个表示容器大小，一个表示初始值（默认为0）, 通常与赋值操作vector[] = value配合。
vector.reserve()只带一个参数，表示容器预留的大小，通常与增添操作vector.insert(&*, value)或vector.push_back(value)配合。

https://leetcode.cn/