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

冷不防 2023-10-04 19:56 24阅读 0赞

描述  
给定一个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;
        }
    }