2364. 统计坏数对的数目 之 灵活思想、学会变形

2364. 统计坏数对的数目

• 看到数据范围，就知道要么使用o(n)的算法进行求解，要么选择o(nlogn)的算法进行求解，首先分析这个条件的形式统计的是j-i != nums[j]-nums[i]的数对的形式，如果暴力进行求解那么对应的时间复杂度会来到o(n^2)
• 思考，原来的形式，对于一个独立的nums[i]来说，由于形式上还与这个j有关，所以求解起来还得每次都得考虑前面的nums[j]的情况,那么有什么好的处理思路？
• 混在一起了，那么就分开来j - nums[j] != i - nums[i]的对数，那么我们又可以转化为 总的情况-等于的情况，这里只需使用到哈希表进行存储即可

from collections import defaultdict
class Solution:def countBadPairs(self, nums: List[int]) -> int:# 移项简化store = defaultdict(int)n = len(nums)ans = 0for i,c in enumerate(nums):ans += store[c-i]store[c-i] += 1return n*(n-1)//2 - ans