力扣-每日一题

最新推荐文章于 2025-04-22 15:22:14 发布

song_ly001

最新推荐文章于 2025-04-22 15:22:14 发布

阅读量672

点赞数 10

文章标签：算法 leetcode 数据结构

本文链接：https://blog.csdn.net/song_ly001/article/details/147320008

版权

如何高效计算数组中坏数对的数量：从暴力解法到最优解

引言

在算法面试和编程竞赛中，数组相关的问题非常常见。今天我们要讨论的是一个关于数组中"坏数对"计数的问题。这个问题看似简单，但通过不同的解法可以很好地展示算法优化的思路。本文将循序渐进地介绍从暴力解法到最优解的思考过程，帮助读者掌握算法优化的技巧。

问题描述

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

示例1：

复制

输入：nums = [4,1,3,3]
输出：5
解释：共有6个数对，其中只有(1,3)是好数对（因为3-1 == 3-1），其余都是坏数对。

示例2：

复制

输入：nums = [1,2,3,4,5]
输出：0
解释：所有数对都是好数对，因为对于任意i<j，都有j-i == nums[j]-nums[i]。

解法一：暴力枚举（不推荐）

思路

最直观的解法是枚举所有可能的数对(i,j)（i<j），然后检查是否满足j-i != nums[j]-nums[i]，统计满足条件的数对数量。

代码实现

java

复制

public long countBadPairs(int[] nums) {
    int n = nums.length;
    long count = 0;
    for (int i = 0; i < n; i++) {
        for (int j = i + 1; j < n; j++) {
            if (j - i != nums[j] - nums[i]) {
                count++;
            }
        }
    }
    return count;
}

复杂度分析

时间复杂度：O(n²)，因为需要双重循环枚举所有数对
空间复杂度：O(1)

缺点

当数组长度较大时（比如n=10⁵），这种解法会非常慢，无法在合理时间内完成计算。

解法二：数学转化+哈希表统计

优化思路

观察坏数对的条件j - i != nums[j] - nums[i]，我们可以将其变形为：

复制

nums[j] - j != nums[i] - i

定义diff[k] = nums[k] - k，那么坏数对就是满足diff[i] != diff[j]的数对。

因此：

坏数对总数 = 总对数 - 好数对总数
总对数 = C(n,2) = n*(n-1)/2
好数对总数 = 所有相同diff值形成的数对数量之和

算法步骤

计算每个元素的diff值：diff[i] = nums[i] - i
统计每个diff值出现的次数
计算好数对总数：对于每个出现m次的diff值，贡献C(m,2)=m*(m-1)/2个好数对
坏数对总数 = 总对数 - 好数对总数

代码实现

java

复制

import java.util.HashMap;
import java.util.Map;

public class Solution {
    public long countBadPairs(int[] nums) {
        int n = nums.length;
        long totalPairs = (long) n * (n - 1) / 2;
        Map<Integer, Integer> freq = new HashMap<>();
        
        // 计算diff并统计频率
        for (int i = 0; i < n; i++) {
            int diff = nums[i] - i;
            freq.put(diff, freq.getOrDefault(diff, 0) + 1);
        }
        
        // 计算好数对的数量
        long goodPairs = 0;
        for (int count : freq.values()) {
            goodPairs += (long) count * (count - 1) / 2;
        }
        
        // 坏数对数量 = 总对数 - 好数对数量
        return totalPairs - goodPairs;
    }
}