LintCode-374.螺旋矩阵

螺旋矩阵

给定一个包含 m x n 个要素的矩阵，（m 行, n 列），按照螺旋顺序，返回该矩阵中的所有要素。

样例

给定如下矩阵：
[
    [ 1, 2, 3 ],
    [ 4, 5, 6 ],
    [ 7, 8, 9 ]
]
应返回 [1,2,3,6,9,8,7,4,5]。

标签

数组 矩阵

code

class Solution {
public:
/**
* @param matrix a matrix of m x n elements
* @return an integer array
*/
vector spiralOrder(vector<vector>& matrix) {
// Write your code here
vector order;;
int tlx,tly,brx,bry;
int i,j;

if(matrix.size() == 0)
return order;
tlx = tly = 0;
brx = matrix[0].size();
bry = matrix.size();

if(brx >= bry) {
for(j=0; j<matrix.size()/2; j++) {
for(i=tlx; i<brx; i++) {
order.push_back(matrix[tlx][i]);
}
for(i=tly+1; i<bry; i++) {
order.push_back(matrix[i][brx-1]);
}
for(i=brx-2; i>=tlx; i--) {
order.push_back(matrix[bry-1][i]);
}
for(i=bry-2; i>=tly+1; i--) {
order.push_back(matrix[i][tlx]);
}
tlx++;tly++; brx--;bry--;
}
if(matrix.size()%2 == 1) {
for(i=tlx; i<brx; i++)
order.push_back(matrix[tlx][i]);
}
}
}
else
{
for(j=0; j<matrix[0].size()/2; j++) {
for(i=tlx; i<brx; i++) {
order.push_back(matrix[tlx][i]);
}
for(i=tly+1; i<bry; i++) {
order.push_back(matrix[i][brx-1]);
}
for(i=brx-2; i>=tlx; i--) {
order.push_back(matrix[bry-1][i]);
}
for(i=bry-2; i>=tly+1; i--) {
order.push_back(matrix[i][tlx]);
}
tlx++;tly++; brx--;bry--;
}
if(matrix[0].size() %2 == 1) {
for(i=tly; i<bry; i++) {
order.push_back(matrix[i][brx-1]);
}
}
}
return order;
}
};