题解 | #k-dis数对#

空间换时间。。。去重有没有更好的办法

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param nums int整型ArrayList 
     * @param k int整型 
     * @return int整型
     */
    public int k_dis (ArrayList<Integer> nums, int k) {
        // write code here
        int n = nums.size();
        Map<Integer, Integer> maps = new HashMap<>();
        int res =0;
        Set<String> set = new HashSet<>();
        for(int i = 0; i < n; i++) {
            if(maps.getOrDefault(nums.get(i)+k, 0) > 0) {
                set.add(""+nums.get(i)+"_"+(nums.get(i)+k));
                set.add(""+(nums.get(i)+k)+"_"+nums.get(i));
            }
            if(maps.getOrDefault(nums.get(i)-k, 0) > 0) {
                set.add(""+nums.get(i)+"_"+(nums.get(i)-k));
                set.add(""+(nums.get(i)-k)+"_"+nums.get(i));
            }
            maps.putIfAbsent(nums.get(i), 1);
        }
        return (set.size()+1)/2;

    }
}