leetcode54.螺旋矩阵

题目大意

给定一个包含 m x n 个元素的矩阵（m 行, n 列），请按照顺时针螺旋顺序，返回矩阵中的所有元素。

解题思路

可以将当前的矩阵看作是多个圈环环嵌套而成的，第一次打印最外面一个矩形，第二次打印第二层的矩形…依次打印即可。
首先用4个变量记录当前打印的圈的左上角和右下角坐标，然后根据坐标约束，先打印上面一行，再打印右边一列，再打印下边一行，最后打印左边一列。

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        if (matrix.size() == 0)
        	return {};
        vector<int> res;
        int row1 = 0, col1 = 0, row2 = matrix.size() - 1, col2 = matrix[0].size() - 1;
        while (row1 <= row2 && col1 <= col2){
        	printCircle(matrix, res, row1++, col1++, row2--, col2--);
        }
        return res;
    }

    //打印每一圈
    void printCircle(vector<vector<int>> & matrix, vector<int> & res, int row1, int col1, int row2, int col2){
    	if (row2 == row1){  //只有一行的情况
    		for (int i = col1; i <= col2; ++i)
    			res.push_back(matrix[row1][i]);
    		return;
    	}
    	if (col1 == col2){  //只有一列的情况 
    		for (int i = row1; i <= row2; ++i)
    			res.push_back(matrix[i][col1]);
    		return ;
    	}

    	int curRow = row1, curCol = col1;
    	//从左到右打印上面一行
    	while (curCol < col2){
    		res.push_back(matrix[curRow][curCol++]);
    	}
    	//从上到下打印右面一列
    	while (curRow < row2){
    		res.push_back(matrix[curRow++][curCol]);
    	}
    	//从右到左打印下面一行
    	while (curCol > col1){
    		res.push_back(matrix[curRow][curCol--]);
    	}
    	//从下到上打印左面一列
    	while (curRow > row1){
    		res.push_back(matrix[curRow--][curCol]);
    	}
    }
};