Re: [閒聊] 每日leetcode

作者: DJYOMIYAHINA (通通打死)   2025-02-09 16:10:06
改算good_pair數目
再用減的
一二三四五
def countBadPairs(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
good_pair_cnt = 0
cnt = defaultdict(int)
cnt[0-nums[0]] = 1
for i in range(1,len(nums)):
if i-nums[i] in cnt:
good_pair_cnt += cnt[i-nums[i]]
cnt[i-nums[i]] += 1
return len(nums)*(len(nums)-1)/2-good_pair_cnt

Links booklink

Contact Us: admin [ a t ] ucptt.com