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

59. Spiral Matrix II

时间:2018-06-11 10:55:40      阅读:163      评论:0      收藏:0      [点我收藏+]

标签:com   矩阵   bre   positive   matrix   并且   from   规律   rect   

问题描述:

Given a positive integer n, generate a square matrix filled with elements from 1 to n2 in spiral order.

Example:

Input: 3
Output:
[
 [ 1, 2, 3 ],
 [ 8, 9, 4 ],
 [ 7, 6, 5 ]
]

解题思路:

这道题我认为主要是找到什么时候变化坐标以及如何变化。

要求螺旋状填充矩阵:即

技术分享图片

可以看到几个关键的变换方向的转折点,那我们来寻找它变换方向的条件:

以ret[i][j]来表达矩阵中的一个元素

当 i > n-1 或 i < 0时需要变换方向,同理对j

当ret[i][j]有值时需要变换方向

可以发现:方向变化都是有规律的

即 右-> 下 -> 左 -> 上 -> 右

不妨以这个顺序来试探,并且改变方向

需要注意的是:

找不到合适方向后需要对当前值进行检测!因为当前值可能就是最后的值!否则会陷入死循环!

代码:

#define R 0
#define D 1
#define L 2
#define U 3

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> ret(n, vector<int>(n, 0));
        int count = 1;
        int i = 0, j = 0;
        int limit = n*n;
        int direction = R;
        while(count <= limit){
            ret[i][j] = count;
            if(direction == R){
                j++;
                if(j < n && ret[i][j] == 0){
                    count++;
                    continue;
                }
                direction = D;
                j--;
                
            }
            if(direction == D){
                i++;
                if(i < n && ret[i][j] == 0){
                    count++;
                    continue;
                }
                direction = L;
                i--;
            }
            if(direction == L){
                j--;
                if(j > -1 && ret[i][j] == 0){
                    count++;
                    continue;
                }
                direction = U;
                j++;
            }
            if(direction == U){
                i--;
                if(i > -1 && ret[i][j] == 0){
                    count++;
                    continue;
                }
                direction = R;
                i++;
            }
            if(count == limit)
                break;
        }
        
        return ret;
    }
};

 

59. Spiral Matrix II

标签:com   矩阵   bre   positive   matrix   并且   from   规律   rect   

原文地址:https://www.cnblogs.com/yaoyudadudu/p/9165321.html

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