Leetcode地址:1995. 统计特殊四元组 - 力扣(LeetCode) (leetcode-cn.com)
哈希法统计有重复元素的情况就有点麻烦,比如[1,1,1,3,5]这种例子,用哈希法很容易少一个解,如果要区分[1,1,1]里面的三个解,对index还需要复杂的辨别
看到一个题解,非常的清楚:【微扰理论】哈希表|两数之和进阶版 - 统计特殊四元组 - 力扣(LeetCode) (leetcode-cn.com)
大体思路:a+b=d-c
时间复杂度是O(n^2)
这里贴出python用dict的实现:
class Solution(object):
def countQuadruplets(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
res=0
cnt={}
for b in range(len(nums)-3,0,-1):
c=b+1
for d in range(c+1,len(nums)):
if nums[d]-nums[c] not in cnt.keys():
cnt[nums[d]-nums[c]]=1
else:
cnt[nums[d]-nums[c]]+=1
for a in range(b):
if nums[a]+nums[b] in cnt.keys():
res+=cnt[nums[a]+nums[b]]
return res