LeetCode热题100 螺旋矩阵

题目描述

给你一个 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

思路

1 定义dx和dy数组代表4个方向。

2 定义times数组代表有n行和m列。

3 每次走完一个方向，会有一行减去或者一列减去。

4 模拟走完这个过程即可。

代码

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
    	int n = matrix.size();
    	int m = matrix[0].size();
    	int dx[] = {0, 1, 0, -1}, dy[]= {1, 0, -1, 0}; // 方向 
    	int times[] = {m, n}; // 行和列的大小 
        vector<int>ans; // 答案
		int x = 0, y = -1, pos = 0 ; // x和y初始化坐标, pos代表方向和行列
        while(ans.size() < n * m)
        {
        	for(int i = 0; i < times[pos % 2]; ++i)
			{
				x += dx[pos];
				y += dy[pos];	
				ans.push_back(matrix[x][y]);
			}	
			pos = (pos + 1) % 4;
			--times[pos % 2] ;
		}    
		return ans; 
	}	
};