LeetCode 2364.统计坏数对的数目

吃着火锅x唱着歌2025-12-01 11:52

给你一个下标从 0 开始的整数数组 nums 。如果 i < j 且 j - i != nums[j] - nums[i] ，那么我们称 (i, j) 是一个坏数对。

请你返回 nums 中坏数对的总数目。

示例 1：

输入：nums = [4,1,3,3]

输出：5

解释：数对 (0, 1) 是坏数对，因为 1 - 0 != 1 - 4 。

数对 (0, 2) 是坏数对，因为 2 - 0 != 3 - 4, 2 != -1 。

数对 (0, 3) 是坏数对，因为 3 - 0 != 3 - 4, 3 != -1 。

数对 (1, 2) 是坏数对，因为 2 - 1 != 3 - 1, 1 != 2 。

数对 (2, 3) 是坏数对，因为 3 - 2 != 3 - 3, 1 != 0 。

总共有 5 个坏数对，所以我们返回 5 。

示例 2：

输入：nums = [1,2,3,4,5]

输出：0

解释：没有坏数对。

提示：

1 <= nums.length <= 10 5 ^5 5

1 <= nums[i] <= 10 9 ^9 9

把j - i != nums[j] - nums[i]变形为nums[i] - i != nums[j] - j，枚举nums，对于当前枚举的值的下标i，我们把nums[i]-i加入哈希表，然后看后面是否还存在j，使得nums[i] - i != nums[j] - j。获取哈希表中不相等的元素数量比较麻烦，可以获取哈希表中相等的元素数量，即好数对数量，然后用总的数对数减去好数对数量

cpp 复制代码

class Solution {
public:
    long long countBadPairs(vector<int>& nums) {
        int n = nums.size();
        long long ans = (long long)(n - 1) * n / 2;

        unordered_map<int, int> m;

        for (int i = 0; i < n; ++i) {
            int target = nums[i] - i;
            if (m.find(target) != m.end()) {
                ans -= m[target];
            }

            ++m[target];
        }

        return ans;
    }
};

如果nums的长度为n，则此算法时间复杂度为O(n)，空间复杂度为O(n)。