题目:
给你一个整数数组 nums 和一个整数 k,请你在数组中找出 不同的 k-diff 数对,并返回不同的 k-diff 数对 的数目。
k-diff 数对定义为一个整数对 (nums[i], nums[j]) ,并满足下述全部条件:
0 <= i, j < nums.length
i != j
nums[i] - nums[j] == k
注意,|val| 表示 val 的绝对值。
示例 1:
输入:nums = [3, 1, 4, 1, 5], k = 2
输出:2
解释:数组中有两个 2-diff 数对, (1, 3) 和 (3, 5)。
尽管数组中有两个 1 ,但我们只应返回不同的数对的数量。
示例 2:
输入:nums = [1, 2, 3, 4, 5], k = 1
输出:4
解释:数组中有四个 1-diff 数对, (1, 2), (2, 3), (3, 4) 和 (4, 5) 。
示例 3:
输入:nums = [1, 3, 1, 5, 4], k = 0
输出:1
解释:数组中只有一个 0-diff 数对,(1, 1) 。
提示:
- 1 <= nums.length <= 104
- -107 <= nums[i] <= 107
- 0 <= k <= 107
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/k-diff-pairs-in-an-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路:
【哈希表】
1.先使用哈希表,对每个数值出现的次数进行计数,对于出现的重复数,出现的次数需要进行累加;
2.然后就需要根据k是否为0进行分情况讨论:
- 当k == 0时,则需要统计数字出现次数大于1的数的个数,检查value值大于1的数的个数即可;
- 当k != 0时,就判断key值中是否有key - k 或key + k存在,统计存在的个数即可。
java代码:
1 class Solution { 2 public int findPairs(int[] nums, int k) { 3 HashMap<Integer, Integer> map = new HashMap<>(); 4 int ans = 0; 5 //给每个值赋上value 6 for(int num : nums){ 7 map.put(num, map.getOrDefault(num, 0) + 1); 8 } 9 for(int key : map.keySet()){ 10 //当k为0,则需要看出现次数大于1的数的个数 11 if(k == 0){ 12 if(map.get(key) > 1) ans++; 13 }else{ 14 //k不为0,就判断key值是否有key+k 15 if(map.containsKey(key + k)){ 16 ans++; 17 } 18 } 19 } 20 return ans; 21 } 22 }
python3代码:
1 class Solution: 2 def findPairs(self, nums: List[int], k: int) -> int: 3 # 创建空字典 4 map = {} 5 ans = 0 6 for num in nums: 7 # 如果当前数字存在于字典的键中,则value+1 8 if num in map: 9 map[num] += 1 10 else: 11 map[num] = 1 12 # 遍历字典的键值 13 for num,value in map.items(): 14 if k == 0: 15 if value > 1: 16 ans += 1 17 else: 18 if num + k in map.keys(): 19 ans += 1 20 return ans标签:map,java,nums,int,数对,力扣,num,diff From: https://www.cnblogs.com/liu-myu/p/16749711.html