力扣每日一题(二十三——统计特殊四元组）

仅以此纪录每日LeetCode所刷题目

题目描述：

示例：

思路一：

奈何本人没文化...一上来就想到了暴力迭代法，不需要什么思路，就是时间复杂度过高

代码一：

class Solution:
    def countQuadruplets(self, nums: List[int]) -> int:
        count = 0
        for i in range(len(nums)-3):
            for j in range(i+1,len(nums)-2):
                for k in range(j+1,len(nums)-1):
                    for w in range(k+1,len(nums)):
                        if nums[i] + nums[j] + nums[k] == nums[w]:
                            count += 1
        return count

思路二：

四重循环我们想办法降成三重循环。

如果我们已经枚举了前三个下标 a, b, ca那么就已经知道了等式左侧 nums[a] + nums[b] + nums[c]的值，即为 nums[d] 的值。对于下标 dd而言，它的取值范围是 c<d<n，那么我们可以使用哈希表统计数组 nums[c+1] 到nums[n−1] 中每个元素出现的次数。这样一来，我们就可以直接从哈希表中获得满足等式的 d的个数，而不需要在 [c+1,n−1] 的范围内进行枚举了。

代码：

class Solution:
    def countQuadruplets(self, nums: List[int]) -> int:
        d = collections.defaultdict(int)
        x = len(nums)
        count = 0
        for i in range(x):
            if nums[i] in d:
                count += d[nums[i]]
            for j in range(i):
                for k in range(j):
                    num = nums[i] + nums[j] + nums[k]
                    d[num] += 1
        return count