今日头条春招第一题

题目描述：

在n个元素的数组中，找到差值为k的数字对去重后的个数。
输入描述：

第一行，n和k，n表示数字个数，k表示差值
第二行，n个正整数
输出描述：

差值为k的数字对去重后的个数
样例

in:
5 2
1 5 3 4 2
out:
3

思路：

使用set存储，去重
排序 遍历查找/二分查找
使用set.contains函数查找是否存在

import java.util.*;
public class Toutiao1 {

    public static void main(String[] args) {
        // TODO Auto-generated method stub
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int k = sc.nextInt();
        int ans = 0;
        int[] a = new int[n];
        List<Integer> numList = new ArrayList<Integer>();
        for(int i = 0; i <n;i++){
            a[i]=sc.nextInt();
            numList.add(a[i]);
        }
        TreeSet<Integer> hset = new TreeSet<Integer>();
        hset.addAll(numList);
        Iterator<Integer> i = hset.iterator();
        while(i.hasNext()){
            Integer integer =  i.next();
            if(hset.contains(integer+k)){
                ans++;
            }
        }
        System.out.println(ans);
    }

}