题目地址:
https://leetcode.com/problems/spiral-matrix/
给定一个矩阵,用spiral的顺序遍历该矩阵,返回结果。
非递归写法可以参考:https://blog.csdn.net/qq_46105170/article/details/108788059。下面介绍递归写法。
递归写法比较好写。每层递归只将最外层加入最终的结果,然后递归地加入内层数字。代码如下:
import java.util.ArrayList;
import java.util.List;
public class Solution {
public List<Integer> spiralOrder(int[][] matrix) {
List<Integer> res = new ArrayList<>();
if (matrix == null || matrix.length == 0 || matrix[0].length == 0) {
return res;
}
addOuter(matrix, 0, 0, matrix.length, matrix[0].length, res);
return res;
}
// 从matrix[x][y]开始将m行n列这个矩阵的最外层加入res
private void addOuter(int[][] matrix, int x, int y, int m, int n, List<Integer> res) {
// 如果矩阵行数或者列数是0,那就不用加了,直接返回
if (m == 0 || n == 0){
return;
}
// 如果只有一行或者一列,那就顺次加入
if (m == 1) {
for (int i = 0; i < n; i++) {
res.add(matrix[x][y + i]);
}
return;
}
if (n == 1) {
for (int i = 0; i < m; i++) {
res.add(matrix[x + i][y]);
}
return;
}
// 把最外层加入res
for (int i = 0; i < n - 1; i++) {
res.add(matrix[x][y + i]);
}
for (int i = 0; i < m - 1; i++) {
res.add(matrix[x + i][y + n - 1]);
}
for (int i = n - 1; i > 0; i--) {
res.add(matrix[x + m - 1][y + i]);
}
for (int i = m - 1; i > 0; i--) {
res.add(matrix[x + i][y]);
}
// 下一层的起点是matrix[x + 1][y + 1],矩阵的行数和列数都减少2
addOuter(matrix, x + 1, y + 1, m - 2, n - 2, res);
}
}
时间复杂度 O ( m n ) O(mn) O(mn),空间复杂度 O ( min ( m / 2 , n / 2 ) ) O(\min(m/2,n/2)) O(min(m/2,n/2))(递归栈深度)。
注:base case一定要写两个, 0 0 0和 1 1 1都要写,因为最后递归求解规模较小问题的时候,问题规模每次是减少 2 2 2的,所以base case要留两个分别应对。