LeetCode 59. Spiral Matrix II

原创于 2017-09-17 21:56:28 发布 · 401 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 同时被 3 个专栏收录

29 篇文章

订阅专栏

LeetCode题解

24 篇文章

订阅专栏

Arr-LeetCode

22 篇文章

订阅专栏

本文介绍了一个算法，该算法接收一个整数 n，并生成一个 n x n 的矩阵，矩阵中的元素从1到n²，按螺旋顺序填充。例如，当 n 为3时，将返回如下矩阵：[[1,2,3],[8,9,4],[7,6,5]]。文章通过一个具体的 C++ 类实现展示了如何实现这一功能。

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 ]
]

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        int tot=1,x=0,y=0;
        vector<vector<int>> v(n,vector<int>(n));
        if(n==1){
            v[0][0]=1;
            return v;
        }
        while(tot<n*n){
            v[0][0]=1;
            while(y+1<n&&!v[x][y+1]){
                v[x][++y]=++tot;
            }
            while(x+1<n&&!v[x+1][y]){
                v[++x][y]=++tot;
            }
            while(y-1>=0&&!v[x][y-1]){
                v[x][--y]=++tot;   
            }
            while(x-1>=0&&!v[x-1][y]){
                v[--x][y]=++tot;
            }
        }
        return v;
    }
};