首页 > 代码库 > [leetcode-59-Spiral Matrix II]

[leetcode-59-Spiral Matrix II]

Given an integer n, generate a square matrix filled with elements from 1 to n2 in spiral order.
For example,
Given n = 3,
You should return the following matrix:
[
[ 1, 2, 3 ],
[ 8, 9, 4 ],
[ 7, 6, 5 ]
]

思路:

4个方向变量上下左右控制边界。

vector<vector<int>> generateMatrix(int n) {
            
         vector<vector<int>>matrix(n,vector<int>(n,0));
         int up = 0, down = n - 1, left = 0, right = n - 1 ,k =1;
         while (1)
         {
             for (int col = left; col <= right; col++)matrix[up][col] = k++;
             if (++up>down)break;
             for (int row = up; row <= down; row++)matrix[row][right] = k++;
             if (--right < left)break;
             for (int col = right; col >= left; col--)matrix[down][col] = k++;
             if (--down < up)break;
             for (int row = down; row >= up; row--)matrix[row][left] = k++;
             if (++left>right)break;
         }
         return matrix; 
    }

 

[leetcode-59-Spiral Matrix II]