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

每日算法之四十一:Spiral Matrix II (螺旋矩阵)

时间:2014-08-15 22:37:39      阅读:274      评论:0      收藏:0      [点我收藏+]

标签:螺旋矩阵

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*4的矩阵,1-12这些数字插入进去是螺旋的一周,用数字标记插入的范围,再转化为和行数、矩阵规模n相关的表达式即可。
第二个问题使用O(n*n)的空间复杂度。
代码如下:
class Solution {
public:
    vector<vector<int> > generateMatrix(int n) {
        vector<vector<int> > matrix;
        if(n < 0)
            return matrix;
        int **m = new int*[n];
        for(int i = 0; i < n; i++)
        {
            m[i] = new int[n];
            memset(m[i], 0, sizeof(int)*n);
        }
        int currentnum = 1;
        int loop = 0;
        while(loop < (n+1)/2)
        {
            int tmp = n - 1 - loop;
            for(int i = loop; i <= tmp; i++)
                m[loop][i] = currentnum++;
            for(int i = loop + 1; i <= tmp; i++)
                m[i][tmp] = currentnum++;
            for(int i = tmp - 1; i >= loop; i--)
                m[tmp][i] = currentnum++;
            for(int i = tmp - 1; i >= loop + 1; i--)
                m[i][loop] = currentnum++;
            loop++;
        }
        for(int i = 0; i < n; i++)
        {
            vector<int> tmp;
            for(int j = 0; j < n; j++)
                tmp.push_back(m[i][j]);
            matrix.push_back(tmp);
        }
        for(int i = 0; i < n; i++)
            delete[] m[i];
        delete[] m;
        return matrix;
    }
};







每日算法之四十一:Spiral Matrix II (螺旋矩阵),布布扣,bubuko.com

每日算法之四十一:Spiral Matrix II (螺旋矩阵)

标签:螺旋矩阵

原文地址:http://blog.csdn.net/yapian8/article/details/38590229

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