数组中的 k-diff 数对

/*
 * 题目思路：
 * 1. 使用Java Stream API进行数据处理。
 * 2. 创建一个哈希集合来存储数组中的数字。
 * 3. 使用stream遍历数组，对于每个数字检查它的加上k或减去k的值是否在哈希集合中。
 * 4. 将符合条件的数对加入到已找到的数对集合中。
 * 5. 最后返回已找到的数对集合的大小。
 */
import java.util.HashSet;
import java.util.Set;
import java.util.stream.Collectors;
import java.util.stream.IntStream;
 
public class KDiffPairsStream {
    public int findPairs(int[] nums, int k) {
        if (k < 0) return 0;
        Set<Integer> numSet = new HashSet<>();
        Set<Integer> foundPairs = IntStream.of(nums)
                .filter(num -> {
                    boolean found = numSet.contains(num - k) || numSet.contains(num + k);
                    numSet.add(num);
                    return found;
                })
                .mapToObj(num -> num - k)
                .collect(Collectors.toSet());
        return foundPairs.size();
    }
}