力扣2364

wang__12300

已于 2025-05-01 15:23:49 修改

阅读量394

点赞数 9

文章标签： java

于 2025-04-18 23:55:28 首次发布

本文链接：https://blog.csdn.net/wang__12300/article/details/147342364

版权

记录

2025.4.18

题目：

在这里插入图片描述

思路：

将j-i!=nums[j]-nums[i]变形成j-nums[j]!=i-nums[i]。
遍历数组，用哈希表记录。

解题步骤：

1.初始化哈希表：HashMap<Integer, Integer> mp用于记录每个键值的出现次数。

2.遍历数组：对于每个元素nums[i]，计算其键值nums[i] - i。

3.计算坏对数量：当前元素i可以与前面i个元素中的i - count个元素形成坏对，其中count是当前键值在哈希表中的出现次数。

4.更新哈希表：每次处理完当前元素后，更新哈希表中当前键值的出现次数。

代码：

class Solution {
    public long countBadPairs(int[] nums) {
        HashMap<Integer, Integer> mp = new HashMap<>();
        long res = 0;
        for (int i = 0; i < nums.length; i++) {
            int key = nums[i] - i;
            res += i - mp.getOrDefault(key, 0);
            mp.put(key, mp.getOrDefault(key, 0) + 1);
        }
        return res;
    }
}