码迷,mamicode.com
首页 > 其他好文 > 详细

LeetCode59. 螺旋矩阵 II

时间:2020-06-27 09:59:28      阅读:60      评论:0      收藏:0      [点我收藏+]

标签:int   site   etc   return   gen   code   cto   改变   ted   

技术图片
这题和第54题类似,都是套一个搜索的模板。
用dx和dy表示方向,方向的顺序是先向右,再向下,再向左,再向上,再向右。。。
如果“撞墙”了就需要改变到下一个方向。“撞墙”的判定就是(newX, newY)越界或者已经被访问过。
“撞墙”就需要改变方向,即更新(newX, newY)。

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> res(n, vector<int>(n));
        vector<vector<bool>> visited(n, vector<bool>(n));
        int dx[4] = {0, 1, 0, -1}, dy[4] = {1, 0, -1, 0};
        for(int x = 0, y = 0, direction = 0, cnt = 1; cnt <= n * n; ++cnt) {
            res[x][y] = cnt;
            visited[x][y] = true;
            int newX = x + dx[direction], newY = y + dy[direction];
            if(newX < 0 || newX >= n || newY < 0 || newY >= n || visited[newX][newY] == true) {
                direction = (direction + 1) % 4;
                newX = x + dx[direction], newY = y + dy[direction];
            }
            x = newX, y = newY;
        }
        return res;
    }
};

LeetCode59. 螺旋矩阵 II

标签:int   site   etc   return   gen   code   cto   改变   ted   

原文地址:https://www.cnblogs.com/linrj/p/13196999.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!