数组-532-数组中的K-diff数对

题目:

给定一个整数数组和一个整数 k, 你需要在数组里找到不同的 k-diff 数对。这里将 k-diff 数对定义为一个整数对 (i, j), 其中 i 和 j 都是数组中的数字,且两数之差的绝对值是 k.
 
示例 1:
 
输入: [3, 1, 4, 1, 5], k = 2
输出: 2
解释: 数组中有两个 2-diff 数对, (1, 3) 和 (3, 5)。
尽管数组中有两个1,但我们只应返回不同的数对的数量。
示例 2:
 
输入:[1, 2, 3, 4, 5], k = 1
输出: 4
解释: 数组中有四个 1-diff 数对, (1, 2), (2, 3), (3, 4) 和 (4, 5)。
示例 3:
 
输入: [1, 3, 1, 5, 4], k = 0
输出: 1
解释: 数组中只有一个 0-diff 数对,(1, 1)。
注意:
 
数对 (i, j) 和数对 (j, i) 被算作同一数对。
数组的长度不超过10,000。
所有输入的整数的范围在 [-1e7, 1e7]。
 
思路分析:
这种数对一定考虑map这种数据结构,但是难点在于,怎么设置hashmap的K-V值:
对于本题来讲,需要一个很重要的信息,数组中每个元素出现的次数!因为我们把元素值放入map中之后重复的元素就只剩下一个,但是当k=0时判断K-diff对时需要元素出现次数的情况。
所以我们设置key为数组中的元素值,value为元素出现的个数。
代码如下:
class Solution {
    public int findPairs(int[] nums, int k) {
       int count=0;
       if(k<0){
           return count;
       }
       Map<Integer, Integer> map=new HashMap<Integer, Integer>();
        for(int i:nums) {   //将数组中的元素放入map中并且统计出现次数
            map.putIfAbsent(i, 0);  //putIfAbsent:如果key没有value值则直接添加进0,如果有不会覆盖之前的value值
            map.put(i, map.get(i)+1); //put:会从新覆盖掉旧值,添加新的value值
            //其实也可以这样写,来实现一个高效的计数器
            // int cut=map.containsKey(i)?map.get(i):0;
            // map.put(i,cut+1);
        }
        //分成两种情况讨论:k=0时,只需要某个key值出现>1次即可;
        for(Integer j:map.keySet()) {
            if(k==0) {
                if(map.get(j)>1) {  
                    count++;
                }
            }
            else {
                if(map.containsKey(j+k)) {
                    count++;
                }
            }
        }
        return count;
    }
}

 

posted @ 2020-05-13 10:08  Yunus  阅读(117)  评论(0编辑  收藏  举报