LeetCode54-螺旋矩阵

LeetCode54-螺旋矩阵

给定一个包含 m x n 个元素的矩阵(m 行, n 列),请按照顺时针螺旋顺序,返回矩阵中的所有元素。

示例 1:

输入:
[
 [ 1, 2, 3 ],
 [ 4, 5, 6 ],
 [ 7, 8, 9 ]
]
输出: [1,2,3,6,9,8,7,4,5]

示例 2:

输入:
[
  [1, 2, 3, 4],
  [5, 6, 7, 8],
  [9,10,11,12]
]
输出: [1,2,3,4,8,12,11,10,9,5,6,7]

一、思路

还记得那个顺时针旋转90度矩阵的题吗?
LeetCode48-旋转图像

思路一样,使用递归来解决,一层一层的保留矩阵中的数据

C++代码:

class Solution {
public:
	vector<int> ans;
	vector<int> spiralOrder(vector<vector<int>>& matrix) {
		if (matrix.empty())
			return ans;

		int row_begin = 0, col_begin = 0;
		int row_end = matrix.size() - 1, col_end = matrix[0].size() - 1;
		int i, j;
		
		while (row_begin < row_end && col_begin < col_end) {
			i = row_begin;
			for (j = col_begin; j <= col_end; j++)
				ans.push_back(matrix[i][j]);

			j = col_end;
			for (i = row_begin + 1; i <= row_end; i++)
				ans.push_back(matrix[i][j]);

			i = row_end;
			for (j = col_end - 1; j >= col_begin; j--)
				ans.push_back(matrix[i][j]);

			j = col_begin;
			for (i = row_end - 1; i > row_begin; i--)
				ans.push_back(matrix[i][j]);

			row_begin++;
			row_end--;
			col_begin++;
			col_end--;
		}

		if (row_begin == row_end) {
			for (j = col_begin; j <= col_end; j++)
				ans.push_back(matrix[row_begin][j]);
		}
		else if (col_begin == col_end) {
			for (i = row_begin; i <= row_end; i++)
				ans.push_back(matrix[i][col_begin]);
		}
		return ans;
	}
};

执行效率:
LeetCode54-螺旋矩阵