Leetcode54. 螺旋矩阵

题目传送地址: ​​https://leetcode.cn/problems/spiral-matrix/​​

运行效率:

代码如下:

public static List<Integer> spiralOrder(int[][] matrix) {
        HashSet<String> set = new HashSet<>();
        int count = matrix.length * matrix[0].length;
        int direction = 0;//走的方向   0:从左往右    1:从上往下  2:从右往左  3:从下往上
        List<Integer> list = new ArrayList<>();
        int rowIndex = 0;
        int colIndex = 0;
        while (list.size() < count) {
            int num = matrix[rowIndex][colIndex];
            list.add(num);
            String coordinate = rowIndex + ":" + colIndex;//坐标
            set.add(coordinate);
            // 校验当前坐标的下一个同方向的坐标是否越界
            boolean validNextCoordinate = validNextCoordinate(direction, coordinate, set, matrix);
            //从左往右
            if (direction == 0) {
                if (validNextCoordinate) {  //如果没有越界，那就继续往右边移动
                    colIndex++;
                } else {   //如果越界，则改变方向
                    direction = 1;
                    rowIndex++;  //改变方向，则下一个位置的坐标应该在当前坐标的下方
                }
                continue;
            }
            //从上往下
            if (direction == 1) {
                if (validNextCoordinate) {  //如果没有越界，那就继续往下边移动
                    rowIndex++;
                } else {   //如果越界，则改变方向 那接下来就是要往左走了
                    direction = 2;
                    colIndex--;
                }
                continue;
            }

            //从右往左
            if (direction == 2) {
                if (validNextCoordinate) {  //如果没有越界，那就继续往左边移动
                    colIndex--;
                } else {   //如果越界，则改变方向 那接下来就是要往上走了
                    direction = 3;
                    rowIndex--;
                }
                continue;
            }
            //从下往上
            if (direction == 3) {
                if (validNextCoordinate) {  //如果没有越界，那就继续往上边移动
                    rowIndex--;
                } else {   //如果越界，则改变方向 那接下来就是要往右走了
                    direction = 0;
                    colIndex++;
                }
            }
        }
        return list;
    }

    /**
     * 校验当前坐标的下一个同方向的坐标是否越界
     *
     * @param direction
     * @param coordinate
     * @param set
     * @param matrix
     * @return
     */
    public static boolean validNextCoordinate(Integer direction, String coordinate, HashSet<String> set, int[][] matrix) {
        String[] split = coordinate.split(":"); //当前坐标位置
        int row = Integer.parseInt(split[0]);  //当前行坐标位置
        int col = Integer.parseInt(split[1]);  //当前列坐标位置
        if (direction == 0) {  //如果是从左到右
            col++;   //行坐标不变，列坐标右移
        }
        if (direction == 1) {  //如果是从上到下
            row++;   //列坐标不变，行坐标下移
        }
        if (direction == 2) {  //如果是从右往左
            col--;   //行坐标不变，列坐标左移
        }
        if (direction == 3) {  //如果是从下往上
            row--;   //列坐标不变，行坐标上移
        }
        //首先要确保下一个位置的坐标不能超过matrix的范围吧
        if (row >= matrix.length || row < 0 || col >= matrix[0].length || col < 0) {
            return false;
        }
        //其次下一个位置的坐标不能在set里面
        String nextCoordinate = row + ":" + col;
        return !set.contains(nextCoordinate);
    }