算法练习-螺旋矩阵（思路+流程图+代码）

本文介绍: 给定一个正整数n，生成一个包含1到 n^2 所有元素，且元素按【顺时针】顺序螺旋排列的正方形矩阵。示例1:输入：n=3输出：[[1,2,3]，[8,9,4]，[7,6,5]]

难度：中等

分类：数组

难度与分类由我所参与的培训课程提供，但需要注意的是，难度与分类仅供参考。以下内容均为个人笔记，旨在督促自己认真学习。

        给定一个正整数n，生成一个包含1到 n^2 所有元素，且元素按【顺时针】顺序螺旋排列的正方形矩阵。
示例1:
        输入：n=3
        输出：[[1,2,3]，[8,9,4]，[7,6,5]]

题目要求生成一个顺时针螺旋排列的正方形矩阵，矩阵元素从1到n^2逐个递增。

1.初始化矩阵：matrix = [[0, 0, 0], [0, 0, 0], [0, 0, 0]]，

matrix = [
  [0, 0, 0],
  [0, 0, 0],
  [0, 0, 0]
]

初始边界：top=0, bottom=2, left=0, right=2。

2.从左到右：matrix[0][0]=1, matrix[0][1]=2, matrix[0][2]=3，

matrix = [
  [1, 2, 3],
  [0, 0, 0],
  [0, 0, 0]
]

更新top=1，

此时，top=1, bottom=2, left=0, right=2。

3.从上到下：matrix[1][2]=6, matrix[2][2]=9，

matrix = [
  [1, 2, 3],
  [0, 0, 4],
  [0, 0, 5]
]

更新right=1，

此时，top=1, bottom=2, left=0, right=1。

4.从右到左：matrix[2][1]=8, matrix[2][0]=7，

matrix = [
  [1, 2, 3],
  [0, 0, 4],
  [7, 6, 5]
]

更新bottom=1，

此时，top=1, bottom=1, left=0, right=1。

5.从下到上：matrix[1][0]=4，

matrix = [
  [1, 2, 3],
  [8, 9, 4],
  [7, 6, 5]
]

更新left=1，

此时，top=1, bottom=1, left=1, right=1。

#include <iostream>
#include <vector>

using namespace std;

// 生成顺时针螺旋矩阵
vector<vector<int>> generateMatrix(int n) {
    // 初始化矩阵
    vector<vector<int>> matrix(n, vector<int>(n, 0));

    // 设置边界指针
    int top = 0, bottom = n - 1, left = 0, right = n - 1;
    int num = 1; // 用于填充矩阵的数字

    // 循环填充矩阵
    while (top <= bottom && left <= right) {
        // 从左到右
        for (int i = left; i <= right; ++i) {
            matrix[top][i] = num++;
        }
        top++;

        // 从上到下
        for (int i = top; i <= bottom; ++i) {
            matrix[i][right] = num++;
        }
        right--;

        // 从右到左
        if (top <= bottom) {
            for (int i = right; i >= left; --i) {
                matrix[bottom][i] = num++;
            }
            bottom--;
        }

        // 从下到上
        if (left <= right) {
            for (int i = bottom; i >= top; --i) {
                matrix[i][left] = num++;
            }
            left++;
        }
    }

    return matrix;
}

// 辅助函数：打印矩阵
void printMatrix(const vector<vector<int>>& matrix) {
    for (const auto& row : matrix) {
        for (int num : row) {
            cout << num << " ";
        }
        cout << endl;
    }
}

int main() {
    int n = 3;
    
    // 生成螺旋矩阵
    vector<vector<int>> result = generateMatrix(n);

    cout << "生成的螺旋矩阵：" << endl;
    
    // 打印矩阵
    printMatrix(result);

    return 0;
}