牛客在线编程101-98 螺旋矩阵

描述
给定一个m x n大小的矩阵（m行，n列），按螺旋的顺序返回矩阵中的所有元素。

数据范围：0 \le n,m \le 100≤n,m≤10，矩阵中任意元素都满足 |val| \le 100∣val∣≤100
要求：空间复杂度 O(nm)O(nm) ，时间复杂度 O(nm)O(nm)

import java.util.ArrayList;
public class Solution {
    public ArrayList<Integer> spiralOrder(int[][] matrix) {
        ArrayList<Integer> res = new ArrayList<Integer>();

        int m = matrix.length;
        if (m == 0) {
            return res;
        }
        int n = matrix[0].length;
        if (n == 0) {
            return res;
        }
        getForstRow(matrix, res);
        return res;
    }

    //每次都取第一行，剩下的数组翻转后继续取
    public int[][] getForstRow(int[][] matrix, ArrayList<Integer> res) {
        if (matrix.length > 1) {
            for (int i = 0; i < matrix[0].length; i++) {
                res.add(matrix[0][i]);
            }
            int m = matrix.length;
            int n = matrix[0].length;
            int[][] array = new int[m - 1][n];
            for (int i = 1; i < m; i++) {
                for (int j = 0; j < n; j++) {
                    array[i - 1][j] = matrix[i][j];
                }
            }
            return getForstRow(xuanzhuan(array), res);
        } else if (matrix.length == 1) {
            for (int i = 0; i < matrix[0].length; i++) {
                res.add(matrix[0][i]);
            }
            return null;
        } else {
            return null;
        }
    }

    public int[][] xuanzhuan(int[][] matrix) {
        //逆时针旋转90°
        int m = matrix.length;
        int n = matrix[0].length;
        int[][] res = new int[n][m];
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                res[n - 1 - j][i] = matrix[i][j];
            }
        }
        return res;
    }
}