【leetcode刷题笔记】Spiral Matrix

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

For example,
Given the following matrix:

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

You should return [1,2,3,6,9,8,7,4,5].

---

题解：还是找规律的题：设有四个变量Xs，Xe，Ys，Ye，如下图所示，它们分别代表当前还未处理的子矩阵的边界，初始为图一所示；运算过程的某一时刻如图二所示

那么螺旋输出就是不停的重复以下四个步骤,直到矩阵所有的元素被输出：

1.输出matrix[Xs->Xe][Ys],Ys++  即1，2，3

2.输出matrix[Xe][Ys->Ye],Xe--  即6,9

3.输出matrix[Xe->Xs][Ye],Ye--  即8,7

4.输出matrix[Ye->Ys][Xs],Xs++  即4

代码如下：

#include<iostream>
#include<vector>
using namespace std;
class Solution {
public:
    vector<int> spiralOrder(vector<vector<int> > &matrix) {
        vector<int> ans;

        if(matrix.size() == 0)
            return ans;

        int Xs = 0,Xe = matrix[0].size()-1,Ys = 0,Ye = matrix.size()-1;
        int total = (Xe+1) * (Ye+1);

        for(int i = 0;i <= total;){
            for(int k = Xs;k <= Xe;k++)
            {
                ans.push_back(matrix[Ys][k]);
                i++;
            }
            if(i >= total)
                break;
            else
                Ys++;

            for(int k = Ys;k <= Ye;k++)
            {
                ans.push_back(matrix[k][Xe]);
                i++;
            }
            if(i >= total)
                break;
            else
                Xe--;

            for(int k = Xe;k >= Xs;k--)
            {
                ans.push_back(matrix[Ye][k]);
                i++;
            }
            if(i >= total)
                break;
            else
                Ye--;

            for(int k = Ye;k >= Ys;k--)
            {
                ans.push_back(matrix[k][Xs]);
                i++;
            }
            if(i >= total)
                break;
            else
                Xs++;
        }
        return ans;
    }
};
int main(){
    Solution s;
    vector<vector<int> > m;
    vector<int>temp(1,1);
    m.push_back(temp);

    vector<int> ans(s.spiralOrder(m));
    cout <<ans.size()<<endl;
}

这里学习到了怎么接受函数返回回来的向量，直接定义新的向量初始化为函数返回值就可以了，如代码65行所示。