所有单元格的远离程度之和

/*
 * 思路:
 * 使用Java Stream API来实现相同的功能。
 * 我们仍然遍历网格中的每一个单元格，然后使用Stream的sorted方法进行排序。
 */
import java.util.stream.*;
import java.util.*;

public class Solution {
    public int[][] allCellsDistOrder(int R, int C, int r0, int c0) {
        return IntStream.range(0, R * C)
                        .mapToObj(i -> new int[]{i / C, i % C})
                        .sorted(Comparator.comparingInt(a -> Math.abs(a[0] - r0) + Math.abs(a[1] - c0)))
                        .toArray(int[][]::new);
    }
}