LeetCode 54 Spiral Matrix

题意：

给出一个数字矩阵，要求从左上角开始蛇形输出。

思路：

可以这样想，对于一个矩阵，每次输出都是它最外层的一圈，迭代着做。即从当前圈的左上角先到右上角再到右下角最后到左下角。

代码：

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>> &matrix) {
        vector<int> ans;
        int rows = matrix.size();
        if (rows == 0) {
            return ans;
        }
        int columns = matrix[0].size();
        for (int i = 0; rows > 0 && columns > 0; ++i, rows -= 2, columns -= 2) {
            if (rows == 1) {
                for (int j = 0; j < columns; ++j) {
                    ans.push_back(matrix[i][j + i]);
                }
            } else if (columns == 1) {
                for (int j = 0; j < rows; ++j) {
                    ans.push_back(matrix[i + j][i]);
                }
            } else {
                for (int j = 0; j < columns; ++j) {
                    ans.push_back(matrix[i][i + j]);
                }
                for (int j = 1; j < rows - 1; ++j) {
                    ans.push_back(matrix[i + j][i + columns - 1]);
                }
                for (int j = columns - 1; j >= 0; --j) {
                    ans.push_back(matrix[i + rows - 1][i + j]);
                }
                for (int j = rows - 2; j >= 1; --j) {
                    ans.push_back(matrix[i + j][i]);
                }
            }
        }
        return ans;
    }
};