对角线遍历

/*
 * 思路：
 * 使用Java Stream API处理数据。
 * 我们沿着对角线遍历矩阵，交替遍历向上和向下的对角线。
 * 当总和为偶数时，我们向上遍历，当总和为奇数时，我们向下遍历。
 * 遍历的过程中，我们将矩阵中的元素加入结果列表中。
 */
import java.util.stream.Collectors;
import java.util.stream.IntStream;
import java.util.ArrayList;
import java.util.List;
 
public class DiagonalTraverseStream {
    public int[] findDiagonalOrder(int[][] mat) {
        int m = mat.length;
        int n = mat[0].length;
        List<Integer> result = new ArrayList<>();
        for (int d = 0; d < m + n - 1; d++) {
            final int sum = d;
            List<Integer> diagonal = IntStream.range(0, m)
                    .filter(i -> sum - i >= 0 && sum - i < n)
                    .mapToObj(i -> mat[i][sum - i])
                    .collect(Collectors.toList());
            if (d % 2 == 0) {
                result.addAll(diagonal);
            } else {
                for (int i = diagonal.size() - 1; i >= 0; i--) {
                    result.add(diagonal.get(i));
                }
            }
        }
        return result.stream().mapToInt(i -> i).toArray();
    }
}