数组-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:
输出: 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:
输出: 4
解释: 数组中有四个 1-diff 数对, (1, 2), (2, 3), (3, 4) 和 (4, 5)。
示例 3:
输入: [1, 3, 1, 5, 4], k = 0
输出: 1
解释: 数组中只有一个 0-diff 数对,(1, 1)。
注意:
输出: 1
解释: 数组中只有一个 0-diff 数对,(1, 1)。
注意:
数对 (i, j) 和数对 (j, i) 被算作同一数对。
数组的长度不超过10,000。
所有输入的整数的范围在 [-1e7, 1e7]。
数组的长度不超过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; } }