[leetcode] Spiral Matrix

Given a matrix of m x n elements (m rows, n columns), return all elements of the matrix in spiral order.

For example,
Given the following matrix:

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

You should return [1,2,3,6,9,8,7,4,5].

参考链接：http://www.cnblogs.com/TenosDoIt/p/3774747.html

思路：

打印螺旋矩阵

逐个环的打印, 对于m *n的矩阵,环的个数是 (min(n,m)+1) / 2。对于每个环顺时针打印四条边。

注意的是：最后一个环可能只包含一行或者一列数据

代码：

class Solution {public:    vector<int> spiralOrder(vector<vector<int> > &matrix) {        int m=matrix.size(),n;        if(m!=0) n=matrix[0].size();        int cycle=m>n?(n+1)/2:(m+1)/2;        vector<int> res;        int a=n,b=m;        for(int i=0;i<cycle;i++,a-=2,b-=2){            for(int column=i;column<i+a;column++){                res.push_back(matrix[i][column]);            }            for(int row=i+1;row<i+b;row++){                res.push_back(matrix[row][i+a-1]);            }            if(a==1 || b==1) break;            for(int column=i+a-2;column>=i;column--){                res.push_back(matrix[i+b-1][column]);            }            for(int row=i+b-2;row>i;row--){                res.push_back(matrix[row][i]);            }        }        return res;    }};