这是悦乐书的第254次更新，第267篇原创
01 看题和准备
今天介绍的是LeetCode算法题中Easy级别的第121题（顺位题号是532）。给定一个整数数组和一个整数k，您需要找到数组中唯一的k-diff对的数量。 这里k-diff对被定义为整数对（i，j），其中i和j都是数组中的数字，它们的绝对差是k。例如：
输入：[3,1,4,1,5]，k = 2
输出：2 说明：数组中有两个2-diff对，（1,3）和（3,5）。虽然我们在输入中有两个1，但我们应该只返回唯一对的数量。输入：[1,2,3,4,5]，k = 1
输出：4 说明：数组中有四个1-diff对，（1,2），（2,3），（3,4）和（4,5）。输入：[1,3,1,5,4]，k = 0
输出：1 说明：数组中有一个0-diff对，（1,1）。注意：
对（i，j）和（j，i）计为同一对。
数组的长度不会超过10,000。
给定输入中的所有整数都属于以下范围：[-1e7, 1e7]。
本次解题使用的开发工具是eclipse，jdk使用的版本是1.8，环境是win7 64位系统，使用Java语言编写和测试。
02 第一种解法
暴力解法。先排序，然后使用两层循环，计算不同元素的绝对值，如果等于k，次数就加1。在外面第一层循环那里，如果前后元素相同，就跳过当前循环，进行下一次循环。在内层循环那里同样做了类似的判断，排除重复计算。
此解法的时间复杂度是O(n^2)，空间复杂度是O(1)。
public int findPairs(int[] nums, int k) { if (nums == null || nums.length == 0 || k < 0) { return 0; } Arrays.sort(nums); int count = 0; for (int i=0; i = 1 && nums[i-1] == nums[i]) { continue; } for (int j=i+1; j = i+2 && nums[j-1] == nums[j]) { continue; } if (Math.abs(n - nums[j]) == k) { count++; } } } return count;} 03 第二种解法
第一种解法时间复杂度太高了，得降低点。第一种解法，我们是做减法，求绝对值，来判断是否等于k，我们也可以做加法，拿当前元素加上k，然后看新元素是否存在于数组中。同时还要考虑重复的计算数据，因此参与计算的元素是唯一的，对此我们可以使用HashMap，已元素值作为key，该元素值出现次数为value。遍历key，如果key加上k后的值存在于map中，次数加1，另外如果k为0的时候，只需要判断每个key所对应的value是否大于等于2即可。
此解法的时间复杂度是O(n)，最坏情况也可能是O(n^2)，空间复杂度是O(n)。
public int findPairs2(int[] nums, int k) { if (nums == null || nums.length == 0 || k < 0) { return 0; } Map map = new HashMap (); for (int n : nums) { map.put(n, map.getOrDefault(n, 0)+1); } int count = 0; if (k == 0) { for (Integer key: map.keySet()) { if (map.get(key) >= 2) { count++; } } } else { for (Integer key: map.keySet()) { if (map.containsKey(key+k)) { count++; } } } return count;} 04 第三种解法
对于第二种解法，还可以将判断放在循环体里面。
public int findPairs3(int[] nums, int k) { if (nums == null || nums.length == 0 || k < 0) { return 0; } Map map = new HashMap (); for (int n : nums) { map.put(n, map.getOrDefault(n, 0)+1); } int count = 0; for (Map.Entry entry : map.entrySet()) { if (k == 0) { if (entry.getValue() >= 2) { count++; } } else { if (map.containsKey(entry.getKey()+k)) { count++; } } } return count; } 05 第四种解法
使用HashSet。使用两个HashSet，同样分为两种情况：k等于0和K不等于0。
如果k等于0时，对数组进行遍历，如果当前元素不存在于set1中，就添加进set1，如果存在set1中，就去判断是否存在于set2中，如果不存在，次数就加1，并将元素添加进set2中。
如果k不等于0，遍历数组，将当前元素添加进set1，将当前元素加上k后再添加进set2，然后使用retainAll方法，将set1中不包含set2元素的元素剔除掉（也就是两set的交集），最后count等于set1中元素的个数。
public int findPairs4(int[] nums, int k) { if (nums == null || nums.length == 0 || k < 0) { return 0; } Set set1 = new HashSet (); Set set2 = new HashSet (); int count = 0; if (k == 0) { for (int n : nums) { if (!set1.contains(n)) { set1.add(n); } else { if (!set2.contains(n)){ count++; } set2.add(n); } } } else { for (int n : nums) { set1.add(n); set2.add(n + k); } set1.retainAll(set2); count = set1.size(); } return count;} 06 第五种解法
使用双指针。还是先将数据排序，定义左右两个指针，分别从0开始，如果左右指针相等，说明是循环的第一次或者重复了，就需要将右指针往后移动一位。如果左指针所指向元素加上k后等于右指针的元素，那么次数加1，接着要判断，如果右指针所指向位置后面的元素和当前元素相等，那么右指针继续往后移动。如果左指针所指向元素加上k后小于右指针的元素，说明左边的元素小了，左指针向前移动。如果左指针所指向元素加上k后大于右指针的元素，说明右边的元素小了，右指针向前移动。
此解法的时间复杂度是O(n log(n))，空间复杂度是O(1)。
public int findPairs5(int[] nums, int k) { if (nums == null || nums.length == 0 || k < 0) { return 0; } Arrays.sort(nums); int count = 0; int start = 0, end = 0; while (end < nums.length) { if (start == end) { end++; } else if (nums[start] + k == nums[end]) { count++; while (end + 1 < nums.length && nums[end] == nums[end + 1]) { end++; } end++; } else if (nums[start] + k < nums[end]) { start++; } else if (nums[start] + k > nums[end]) { end++; } } return count;} 07 小结
此题的测试用例中，k出现了负值，所以在特殊情况判断中，还需要判断k小于0，这也是本题不严谨的一个地方。
算法专题目前已日更超过三个月，算法题文章121+篇，公众号对话框回复【数据结构与算法】、【算法】、【数据结构】中的任一关键词，获取系列文章合集。
以上就是全部内容，如果大家有什么好的解法思路、建议或者其他问题，可以下方留言交流，点赞、留言、转发就是对我最大的回报和支持！