1 题目:螺旋矩阵
官方标定难度:中
给你一个 m 行 n 列的矩阵 matrix ,请按照 顺时针螺旋顺序 ,返回矩阵中的所有元素。
示例 1:
输入:matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出:[1,2,3,6,9,8,7,4,5]
示例 2:
输入:matrix = [[1,2,3,4],[5,6,7,8],[9,10,11,12]]
输出:[1,2,3,4,8,12,11,10,9,5,6,7]
提示:
m == matrix.length
n == matrix[i].length
1 <= m, n <= 10
-100 <= matrix[i][j] <= 100
2 solution
采用模拟法,模拟螺旋的顺序填数字,确定好方向和切换方向的时机,详见代码
代码
class Solution {
public:
vector<int> spiralOrder(vector<vector<int>> &matrix) {vector<int> result;if(matrix[0].size() == 1){for(auto &s:matrix) result.push_back(s[0]);return result;}int dr[] = {0, 1, 0, -1};int dc[] = {1, 0, -1, 0};int p = 0;int m = matrix.size();int n = matrix[0].size();int sm = 1, sn = 0;int N = n * m;m--, n--;int r = 0, c = 0;for (int i = 0; i < N; i++) {result.push_back(matrix[r][c]);r += dr[p];c += dc[p];if (p == 0 && c == n) {n--;p = (p + 1) % 4;}else if (p == 1 && r == m) {m--;p = (p + 1) % 4;}else if (p == 2 && c == sn) {sn++;p = (p + 1) % 4;}else if (p == 3 && r == sm) {sm++;p = (p + 1) % 4;}}return result;
}};