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.

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].

 

---

题目标签：Array

　　这道题目给了我们一个矩阵，m*n，让我们返回一个螺旋矩阵排列的array，这题目名字听着挺酷炫呀。我们来看原题例子：

　　1  2  3

　　4  5  6

　　7  8  9

　　一共有4步骤：我们还需要设置rowBegin = 0, rowEnd = 2, colBegin = 0, colEnd = 2.

　　1. 红色：从rowBegin这一排向右遍历，加入每一个matrix[rowBegin][j],   j: colBegin ~ colEnd. 遍历完要把rowBegin++, 从下一排继续；

　　2. 绿色：从colEnd这一列开始向下遍历，加入每一个matrix[j][colEnd], j: rowBegin ~ rowEnd. 遍历完要把colEnd--, 从左边那列继续；

　　3. 蓝色：从rowEnd这一排向左遍历，加入每一个matrix[rowEnd][j], j: colEnd ~ colBegin. 遍历完要把rowEnd--, 从上一排继续。

　　4. 灰色：从colBegin这一列开始向上遍历，加入每一个matrix[j][colBegin], j: rowEnd ~ rowBegin, 遍历完要把colBegin++, 从右边那列继续。

 

　　那什么时候要结束呢，可以利用m*n 得到一共的数量，每次加入一个，就把这个数量-1， 当数量等于0的时候意味着我们加完了所有的数字，return res 就可以了。

 

 

Java Solution:

Runtime beats 20.42% 

完成日期：07/18/2017

关键词：Array

关键点：螺旋矩阵的固定模式

 

public class Solution 
{
    public List<Integer> spiralOrder(int[][] matrix) 
    {
        List<Integer> res = new ArrayList<Integer>();
        
        if(matrix.length == 0)
            return res;
        
        int totalNum = matrix.length * matrix[0].length;
        
        int rowBegin = 0;
        int rowEnd = matrix.length - 1;
        int colBegin = 0;
        int colEnd = matrix[0].length - 1;
        
        while(totalNum > 0)
        {
            // traverse right
            for(int j=colBegin; j<=colEnd && totalNum>0; j++)
            {
                res.add(matrix[rowBegin][j]);
                totalNum--;
            }
            rowBegin++;
            
            // traverse Down
            for(int j=rowBegin; j<=rowEnd && totalNum>0; j++)
            {
                res.add(matrix[j][colEnd]);
                totalNum--;
            }
            colEnd--;
            
            // traverse left
            for(int j=colEnd; j>=colBegin && totalNum>0; j--)
            {
                res.add(matrix[rowEnd][j]);
                totalNum--;
            }
            rowEnd--;
            
            // traverse up
            for(int j=rowEnd; j>= rowBegin && totalNum>0; j--)
            {
                res.add(matrix[j][colBegin]);
                totalNum--;
            }
            colBegin++;
        }
        
        return res;
        
    }
}

参考资料：

https://leetcode.com/problems/spiral-matrix/#/discuss

 

LeetCode 算法题目列表 - LeetCode Algorithms Questions List