不邻接植花

/*
思路：
1. 使用Java Stream进行流式操作来生成答案。
2. 和Java的普通解法一样，使用贪心算法选取每个花园的花。
*/

import java.util.*;
import java.util.stream.IntStream;

public class GardenNoAdjStream {
    public int[] gardenNoAdj(int n, int[][] paths) {
        int[] answer = new int[n];
        List<Integer>[] graph = IntStream.range(0, n).mapToObj(i -> new ArrayList<Integer>()).toArray(List[]::new);
        Arrays.stream(paths).forEach(path -> {
            graph[path[0] - 1].add(path[1] - 1);
            graph[path[1] - 1].add(path[0] - 1);
        });
        IntStream.range(0, n).forEach(i -> {
            boolean[] used = new boolean[5];
            graph[i].forEach(neighbor -> used[answer[neighbor]] = true);
            answer[i] = IntStream.range(1, 5).filter(j -> !used[j]).findFirst().getAsInt();
        });
        return answer;
    }
}