19.2.8 [LeetCode 54] Spiral Matrix

Given a matrix of m x n elements (m rows, n columns), return all elements of the matrix in spiral order.

Example 1:

Input:
[
 [ 1, 2, 3 ],
 [ 4, 5, 6 ],
 [ 7, 8, 9 ]
]
Output: [1,2,3,6,9,8,7,4,5]

Example 2:

Input:
[
  [1, 2, 3, 4],
  [5, 6, 7, 8],
  [9,10,11,12]
]
Output: [1,2,3,4,8,12,11,10,9,5,6,7]

题意

顺时针螺旋输出数组

题解

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        if (matrix.empty())return vector<int>();
        int line=matrix.size(),row=matrix[0].size(),n = row*line;
        vector<int>ans;
        int s = -1, e = row, ss = -1, ee = line;
        for (int i = 0; i < (line + 1) / 2; i++) {
            s++, e--, ss++, ee--;
            if (e < s || ee < ss)break;
            for (int j = s; j <= e; j++)
                ans.push_back(matrix[i][j]);
            for (int j = ss+1; j <= ee; j++)
                ans.push_back(matrix[j][e]);
            if (ss != ee) {
                for (int j = e - 1; j >= s; j--)
                    ans.push_back(matrix[ee][j]);
            }
            if (s != e) {
                for (int j = ee - 1; j > ss; j--)
                    ans.push_back(matrix[j][s]);
            }
        }
        return ans;
    }
};

比较容易WA，要处理好特殊情况