LeetCode 59 Spiral Matrix II

来源:互联网 发布:overlay 网络 编辑:程序博客网 时间:2024/06/04 19:21

题意:

画出n*n的蛇形矩阵。


思路:

与 http://blog.csdn.net/houserabbit/article/details/72763411 一致,更简单一点,因为它是正方形……

这题还可以扩展,因为如果不是让画出矩阵,只是输出坐标(x,y)的数字是几,那么完全可以不开空间存矩阵,按照矩阵生成方法生成到(x,y)坐标时输出即可。


代码:

class Solution {public:    vector<vector<int>> generateMatrix(int n) {        vector<vector<int>> ans;        for (int i = 0; i < n; ++i) {            ans.push_back(vector<int>(n));        }        int num = 1;        for (int i = 0; i < n / 2; ++i) {            int m = n - i * 2;            for (int j = 0; j < m - 1; ++j) ans[i][i + j] = num++;            for (int j = 0; j < m - 1; ++j) ans[i + j][i + m - 1] = num++;            for (int j = m - 1; j > 0; --j) ans[i + m - 1][i + j] = num++;            for (int j = m - 1; j > 0; --j) ans[i + j][i] = num++;        }        if (n & 1) {            ans[n / 2][n / 2] = num++;        }        return ans;    }};


原创粉丝点击