向数组中追加 K 个整数

/*
 * 思路：
 * 使用Java Stream API实现与上述思路相同的逻辑。
 * 通过生成一个无限的IntStream，过滤掉在nums中出现的数，
 * 然后取前k个数，并计算它们的和。
 */

import java.util.Arrays;
import java.util.Set;
import java.util.stream.Collectors;
import java.util.stream.IntStream;

public class SolutionStream {
    public int minimalSum(int[] nums, int k) {
        Set<Integer> numSet = Arrays.stream(nums).boxed().collect(Collectors.toSet());
        return IntStream.iterate(1, n -> n + 1)
                        .filter(n -> !numSet.contains(n))
                        .limit(k)
                        .sum();
    }
}