LeetCode 59. 螺旋矩阵 II (C++实现)

1. 题目描述

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

示例 1：

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

示例 2：

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

2. 解题思路

首先定义左指针left、右指针right，上指针top，下指针bottom，然后从1开始循环到n^2，遇到边界的时候指针相应地变化。

3. 代码实现

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

        int idx = 1;
        int target = n * n;

        while(idx <= target)
        {
            for (int i = left; i <= right; i++)
            {
                ans[top][i] = idx;
                idx++;
            }
            top++;
            for (int i = top; i <= bottom; i++)
            {
                ans[i][right] = idx;
                idx++;
            }
            right--;
            for (int i = right; i >= left; i--)
            {
                ans[bottom][i] = idx;
                idx++;
            }
            bottom--;
            for (int i = bottom; i >= top; i--)
            {
                ans[i][left] = idx;
                idx++;
            }
            left++;

        }
        return ans;
    }
};