2021-01-25

螺旋矩阵

题目

给定一个包含 m x n 个元素的矩阵（m 行, n 列），请按照顺时针螺旋顺序，返回矩阵中的所有元素。

示例 ：
输入：
[[1, 2, 3, 4],
[5, 6, 7, 8],
[9,10,11,12]]
输出: [1,2,3,4,8,12,11,10,9,5,6,7]

题解

按圈返回，先返回外层圈：
1，2， 3，4
5， ， ， 8
9, 10, 11, 12
外圈返回结果：1，2，3，4，8，12，11，10，9，5

接着返回次外层圈：
6， 7
次外层圈返回结果：6，7

按圈返回代码如下：
//tr表示每层圈起始行数，dr表示末尾行数
//dr表示每层圈起始列数，dc表示末尾列数

public void edge(int[][] m, int tr, int tc, int dr, int dc, List<Integer> list){

    if(tr == dr){ //矩阵只有一行
        for(int i = tc; i <= dc; i++){
            list.add(m[tr][i]);
        }
    }else if(tc == dc){ //矩阵只有一列
        for(int i = tr; i <= dr; i++){
            list.add(m[i][tc]);
        }
    }else{//正常多行多列矩阵
        for(int i = tc; i < dc; i++){
            list.add(m[tr][i]);
        }
        for(int i = tr; i < dr; i++){
            list.add(m[i][dc]);
        }
        for(int i = dc; i > tc; i--){
            list.add(m[dr][i]);
        }
        for(int i = dr; i > tr; i--){
            list.add(m[i][tc]);
        }
    }

全部代码

public List<Integer> spiralOrder(int[][] matrix) {
    List<Integer> list = new ArrayList<>();
    int tr = 0;
    int tc = 0;
    int dr = matrix.length - 1;
    int dc = matrix[0].length - 1;
    while(tr <= dr && tc <= dc){
        edge(matrix, tr++, tc++, dr--, dc--, list);
    }
    return list;
}

public void edge(int[][] m, int tr, int tc, int dr, int dc, List<Integer> list){
    
    if(tr == dr){
        for(int i = tc; i <= dc; i++){
            list.add(m[tr][i]);
        }
    }else if(tc == dc){
        for(int i = tr; i <= dr; i++){
            list.add(m[i][tc]);
        }
    }else{
        for(int i = tc; i < dc; i++){
            list.add(m[tr][i]);
        }
        for(int i = tr; i < dr; i++){
            list.add(m[i][dc]);
        }
        for(int i = dc; i > tc; i--){
            list.add(m[dr][i]);
        }
        for(int i = dr; i > tr; i--){
            list.add(m[i][tc]);
        }
    }