LeetCode（54）：螺旋矩阵

Medium！

题目描述：

给定一个包含 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]

解题思路：

这道题让我们将一个矩阵按照螺旋顺序打印出来，我们只能一条边一条边的打印，首先我们要从给定的mxn的矩阵中算出按螺旋顺序有几个环，注意最终间的环可以是一个数字，也可以是一行或者一列。环数的计算公式是 min(m, n) / 2，知道了环数，我们可以对每个环的边按顺序打印，比如对于题目中给的那个例子，个边生成的顺序是(用颜色标记了数字) Red -> Green -> Blue -> Yellow -> Black：

1　2　3

4　5　6

7　8　9

我们定义p，q为当前环的高度和宽度，当p或者q为1时，表示最后一个环只有一行或者一列，可以跳出循环。此题的难点在于下标的转换，如何正确的转换下标是解此题的关键，我们可以对照着上面的3x3的例子来完成下标的填写。

C++解法一：

 1 class Solution {
 2 public:
 3     vector<int> spiralOrder(vector<vector<int> > &matrix) {
 4         vector<int> res;
 5         if (matrix.empty() || matrix[0].empty()) return res;
 6         int m = matrix.size(), n = matrix[0].size();
 7         int c = m > n ? (n + 1) / 2 : (m + 1) / 2;
 8         int p = m, q = n;
 9         for (int i = 0; i < c; ++i, p -= 2, q -= 2) {
10             for (int col = i; col < i + q; ++col) 
11                 res.push_back(matrix[i][col]);
12             for (int row = i + 1; row < i + p; ++row)
13                 res.push_back(matrix[row][i + q - 1]);
14             if (p == 1 || q == 1) break;
15             for (int col = i + q - 2; col >= i; --col)
16                 res.push_back(matrix[i + p - 1][col]);
17             for (int row = i + p - 2; row > i; --row) 
18                 res.push_back(matrix[row][i]);
19         }
20         return res;
21     }
22 };