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

【Spiral Matrix II】cpp

时间:2015-06-09 08:30:13      阅读:106      评论: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 ]
]

代码:

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
            vector<vector<int> > ret(n, vector<int>(n,0));
            int circle = n/2;
            int v = 1;
            for ( int c=0; c<circle; ++c )
            {
                // up
                for ( int col=c; col<n-c; ++col ) ret[c][col]=v++;
                // right
                for ( int row=c+1; row<n-c-1; ++row ) ret[row][n-1-c]=v++;
                // down
                for ( int col=n-1-c; col>=c; --col ) ret[n-1-c][col]=v++;
                // left
                for ( int row=n-2-c; row>c; --row) ret[row][c]=v++;
            }
            if ( n & 1 )
            {
                ret[circle][circle]=v;
            }
            return ret;
    }
};

tips:

按照Spiral Matrix的顺序走一遍元素,维护一个v每次运算后+1。

【Spiral Matrix II】cpp

标签:

原文地址:http://www.cnblogs.com/xbf9xbf/p/4562464.html

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