LeetCode 54. Spiral Matrix 螺旋矩阵(Java)

题目:

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

Example 1:
Input:
[
[ 1, 2, 3 ],
[ 4, 5, 6 ],
[ 7, 8, 9 ]
]
Output: [1,2,3,6,9,8,7,4,5]

Example 2:
Input:
[
[1, 2, 3, 4],
[5, 6, 7, 8],
[9,10,11,12]
]
Output: [1,2,3,4,8,12,11,10,9,5,6,7]

解答

题目要求为顺时针返回矩阵的所有元素

解题思路为:
分别设定 top、down、left、right 四个边界值,分别指向剩余待遍历的矩阵的上边界、下边界、左边界和右边界。从矩阵左上角第一个元素开始,分别沿右、下、左、上四个方向走,当达到边界条件时,则更改为下一个方向,并缩短对应的边界值

class Solution {
    public List<Integer> spiralOrder(int[][] matrix) {
        List<Integer> res = new ArrayList<>();
        if(matrix.length == 0) {
            return res;
        }
        int row = matrix.length;   //行数
        int colnmn = matrix[0].length;   //列数
        int top = 0;
        int left = 0;
        int right = colnmn-1;
        int down = row-1;
        while(res.size() < row * colnmn) {
            for(int i=left; i<=right; i++) {
                res.add(matrix[top][i]);
            }
            top++;
            if(res.size() < row * colnmn) {
                for(int i=top; i<=down; i++) {
                    res.add(matrix[i][right]);
                }
                right--;
            }
            if(res.size() < row * colnmn) {
                for(int i=right; i>=left; i--) {
                    res.add(matrix[down][i]);
                }
                down--;
            }
            if(res.size() < row * colnmn) {
                for(int i=down; i>=top; i--) {
                    res.add(matrix[i][left]);
                }
                left++;
            }
        }
        return res;
    }
}

注意对if(matrix.length == 0)的判空处理,第一次提交忽略导致错误。

你可能感兴趣的:(LeetCode)