【leetcode刷题笔记】Spiral Matrix

时间:2014-04-29 10:37:46   收藏:0   阅读:434

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:

mamicode.com,码迷
[
 [ 1, 2, 3 ],
 [ 4, 5, 6 ],
 [ 7, 8, 9 ]
]
mamicode.com,码迷

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


题解:还是找规律的题:设有四个变量Xs,Xe,Ys,Ye,如下图所示,它们分别代表当前还未处理的子矩阵的边界,初始为图一所示;运算过程的某一时刻如图二所示

mamicode.com,码迷

那么螺旋输出就是不停的重复以下四个步骤,直到矩阵所有的元素被输出:

1.输出matrix[Xs->Xe][Ys],Ys++  即1,2,3

2.输出matrix[Xe][Ys->Ye],Xe--  即6,9

3.输出matrix[Xe->Xs][Ye],Ye--  即8,7

4.输出matrix[Ye->Ys][Xs],Xs++  即4

代码如下:

mamicode.com,码迷
 1 #include<iostream>
 2 #include<vector>
 3 using namespace std;
 4 class Solution {
 5 public:
 6     vector<int> spiralOrder(vector<vector<int> > &matrix) {
 7         vector<int> ans;
 8 
 9         if(matrix.size() == 0)
10             return ans;
11 
12         int Xs = 0,Xe = matrix[0].size()-1,Ys = 0,Ye = matrix.size()-1;
13         int total = (Xe+1) * (Ye+1);
14 
15         for(int i = 0;i <= total;){
16             for(int k = Xs;k <= Xe;k++)
17             {
18                 ans.push_back(matrix[Ys][k]);
19                 i++;
20             }
21             if(i >= total)
22                 break;
23             else
24                 Ys++;
25 
26             for(int k = Ys;k <= Ye;k++)
27             {
28                 ans.push_back(matrix[k][Xe]);
29                 i++;
30             }
31             if(i >= total)
32                 break;
33             else
34                 Xe--;
35 
36             for(int k = Xe;k >= Xs;k--)
37             {
38                 ans.push_back(matrix[Ye][k]);
39                 i++;
40             }
41             if(i >= total)
42                 break;
43             else
44                 Ye--;
45 
46             for(int k = Ye;k >= Ys;k--)
47             {
48                 ans.push_back(matrix[k][Xs]);
49                 i++;
50             }
51             if(i >= total)
52                 break;
53             else
54                 Xs++;
55         }
56         return ans;
57     }
58 };
59 int main(){
60     Solution s;
61     vector<vector<int> > m;
62     vector<int>temp(1,1);
63     m.push_back(temp);
64 
65     vector<int> ans(s.spiralOrder(m));
66     cout <<ans.size()<<endl;
67 }
mamicode.com,码迷

这里学习到了怎么接受函数返回回来的向量,直接定义新的向量初始化为函数返回值就可以了,如代码65行所示。

 

【leetcode刷题笔记】Spiral Matrix,码迷,mamicode.com

评论(0
© 2014 mamicode.com 版权所有 京ICP备13008772号-2  联系我们:gaon5@hotmail.com
迷上了代码!