【LeetCode】54. Spiral Matrix

问题描述

https://leetcode.com/problems/spiral-matrix/#/description
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].

获取矩阵的螺旋顺序。

算法分析

方法1：

使用一个指针(x,y)从(0,0)开始遍历，将数添加到数组中，每次一撞墙就转向：如果当前是遍历列，就改为遍历行；当前遍历行，就改为遍历列。使用一个布尔数组表示数是否已被遍历，如果已被遍历，其本身就变为墙。

但控制方式比较复杂。

方法2：

参考自：https://discuss.leetcode.com/topic/3713/super-simple-and-easy-to-understand-solution

每次遍历无非是遍历一行或一列，有个开头和结尾，设为rowBegin（开始行），rowEnd（结束行），colBegin（开始列），colEnd（结束列），然后开始循环：

1. 首先从左到右遍历（colBegin->colEnd，结束后就遍历完上面的一行了，rowBegin+1）
2. 然后从上到下遍历（rowBegin->rowEnd，结束后就遍历完右边一列了，colEnd-1）
3. 接着从右到左遍历（colEnd-colBegin，结束后就遍历完下面的一行了，rowEnd-1）
4. 最后从下到上遍历（rowEnd->rowBegin，结束后就遍历完左边的一列了，colBegin+1）

一直到colBegin>colEnd并且rowBegin>rowEnd，循环结束

代码

/**         * 获取矩阵的螺旋顺序。         * 使用一个指针从(0,0)开始遍历，将数添加到数组中，每次一撞墙就转向：如果当前是遍历列，就改为遍历行；当前遍历行，就改为遍历列。         * 使用一个布尔数组表示数是否已被遍历，如果已被遍历，其本身就变为墙。         *          * 但控制方式很复杂。         */        public List<Integer> spiralOrder1(int[][] matrix) {            List<Integer> res = new ArrayList<>();            if(matrix==null||matrix.length == 0) {                return res;            }            int x = 0, y = 0, m = matrix.length, n = matrix[0].length;            boolean[][] visited = new boolean[m][n];            boolean row = false; // 方向控制，false为遍历列，true为遍历行            int delta = 1; // 增量，1或-1            while(true) {                if(res.size() == m * n) {                    break;                }                res.add(matrix[x][y]);                visited[x][y] = true;                if(row) {                    if(delta==1&&x<m-1 || delta==-1&&x>0)                        x+=delta;                    if((delta==1&&(x==m-1||m>1&&visited[x+1][y]||m==1))||(delta==-1&&(x==0||visited[x-1][y]))) {                        row = false; // 转向                        if(y==0||visited[x][y-1]) delta=1;                        else if(y==n-1||visited[x][y+1]) delta=-1;                    }                } else {                    if(delta==1&&y<n-1 || delta==-1&&y>0)                        y+=delta;                    if((delta==1&&(y==n-1||n>1&&visited[x][y+1]||n==1))||(delta==-1&&(y==0||visited[x][y-1]))) {                        row = true; // 转向                        if(x==0||visited[x-1][y]) delta=1;                        else if(x==m-1||visited[x+1][y]) delta=-1;                        if(n==1) x+=delta;                    }                }            }            return res;        }        /**         * 方法2：         * 参考自：https://discuss.leetcode.com/topic/3713/super-simple-and-easy-to-understand-solution         * 每次遍历无非是遍历一行或一列，有个开头和结尾，设为rowBegin（开始行），rowEnd（结束行），colBegin（开始列），colEnd（结束列），然后开始循环：         * 首先从左到右遍历（colBegin->colEnd，结束后就遍历完上面的一行了，rowBegin+1）         * 然后从上到下遍历（rowBegin->rowEnd，结束后就遍历完右边一列了，colEnd-1）         * 接着从右到左遍历（colEnd-colBegin，结束后就遍历完下面的一行了，rowEnd-1）         * 最后从下到上遍历（rowEnd->rowBegin，结束后就遍历完左边的一列了，colBegin+1）         * 一直到colBegin>colEnd并且rowBegin>rowEnd，循环结束         * @param matrix         * @return         */        public List<Integer> spiralOrder(int[][] matrix) {            List<Integer> res = new ArrayList<>();            if(matrix == null || matrix.length==0) {                return res;            }            int rowBegin = 0, rowEnd = matrix.length-1;            int colBegin = 0, colEnd = matrix[0].length-1;            while(rowBegin<=rowEnd && colBegin<=colEnd) {                // 从左到右                for(int i=colBegin;i<=colEnd;i++) {                    res.add(matrix[rowBegin][i]);                }                rowBegin++;                // 从上到下                for(int i=rowBegin;i<=rowEnd;i++) {                    res.add(matrix[i][colEnd]);                }                colEnd--;                // 从右到左                if (rowBegin <= rowEnd) {                    for(int i=colEnd;i>=colBegin;i--) {                        res.add(matrix[rowEnd][i]);                    }                }                rowEnd--;                // 从下到上                if (colBegin <= colEnd) {                    for(int i=rowEnd;i>=rowBegin;i--) {                        res.add(matrix[i][colBegin]);                    }                }                colBegin++;            }            return res;        }