[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].

开一个数组用来保存,之前这个单元是否被使用过,并配合边界判断,DFS,最后得出结果。空间复杂度O(n*m),时间O(n*m)

 1 class Solution {
 2 private:
 3     int step[4][2];
 4     vector<int> ret;
 5     bool canUse[100][100];
 6 public:
 7     void dfs(vector<vector<int> > &matrix, int direct, int x, int y)
 8     {
 9         for(int i = 0; i < 4; i++)
10         {
11             int j = (direct + i) % 4;
12             int tx = x + step[j][0];
13             int ty = y + step[j][1];
14             if (0 <= tx && tx < matrix.size() && 0 <= ty && ty < matrix[0].size() && canUse[tx][ty])
15             {
16                 canUse[tx][ty] = false;
17                 ret.push_back(matrix[tx][ty]);                
18                 dfs(matrix, j, tx, ty);               
19             }            
20         }
21     }
22     
23     vector<int> spiralOrder(vector<vector<int> > &matrix) {
24         // Start typing your C/C++ solution below
25         // DO NOT write int main() function
26         step[0][0] = 0;
27         step[0][1] = 1;
28         step[1][0] = 1;
29         step[1][1] = 0;
30         step[2][0] = 0;
31         step[2][1] = -1;
32         step[3][0] = -1;
33         step[3][1] = 0;
34         ret.clear();
35         memset(canUse, true, sizeof(canUse));
36         dfs(matrix, 0, 0, -1);
37         
38         return ret;
39     }
40 };
posted @ 2012-11-18 11:24  chkkch  阅读(1943)  评论(0编辑  收藏  举报