沙地治理

/*
 * 思路：
 * 1. 使用Java Stream来简化代码结构。
 * 2. 从最底层开始选择绿地，使得上层的沙地尽快转化为绿地。
 */
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.IntStream;

public class SandDesertStream {
    public List<int[]> plantTrees(int size) {
        return IntStream.rangeClosed(1, size).boxed()
                .flatMap(i -> IntStream.rangeClosed(1, 2 * i - 1)
                        .filter(j -> (i % 2 == 1 && j % 2 == 1) || (i % 2 == 0 && j % 2 == 0))
                        .mapToObj(j -> new int[]{i, j}))
                .collect(Collectors.toList());
    }

    public static void main(String[] args) {
        SandDesertStream sd = new SandDesertStream();
        List<int[]> trees = sd.plantTrees(3);
        trees.forEach(tree -> System.out.println("[" + tree[0] + "," + tree[1] + "]"));
    }
}