1995. 统计特殊四元组

Difficulty
easy
Tags
n数之和
URL
https://leetcode-cn.com/problems/count-special-quadruplets/
Star
给你一个 下标从 0 开始 的整数数组 nums ，返回满足下述条件的 不同 四元组 (a, b, c, d) 的 数目 ：
  • nums[a] + nums[b] + nums[c] == nums[d] ，且
  • a < b < c < d
示例 1：
输入：nums = [1,2,3,6]
输出：1
解释：满足要求的唯一一个四元组是 (0, 1, 2, 3) 因为 1 + 2 + 3 == 6 。
示例 2：
输入：nums = [3,3,6,4,5]
输出：0
解释：[3,3,6,4,5] 中不存在满足要求的四元组。
示例 3：
输入：nums = [1,1,1,3,5]
输出：4
解释：满足要求的 4 个四元组如下：
- (0, 1, 2, 3): 1 + 1 + 1 == 3
- (0, 1, 3, 4): 1 + 1 + 3 == 5
- (0, 2, 3, 4): 1 + 1 + 3 == 5
- (1, 2, 3, 4): 1 + 1 + 3 == 5
提示：
  • 4 <= nums.length <= 50
  • 1 <= nums[i] <= 100

法1 暴力枚举

思路
数据量只有50，直接四层循环暴力枚举每一种可能
题解
class Solution:
    def countQuadruplets(self, nums: List[int]) -> int:
        res = 0
        for i in range(0, len(nums)-3):
            for j in range(i+1, len(nums)-2):
                for k in range(j+1, len(nums)-1):
                    for z in range(k+1, len(nums)):
                        if nums[i] + nums[j] + nums[k] == nums[z]:
                            res += 1

        return res

法2 hashmap

思路
nums[a] + nums[b] = nums[d] - nums[c]
所以可以从后往前将nums[d] - nums[c] 存起来，然后遍历选择前面的nums[a],nums[b]
那该怎么将nums[d] - nums[c] 存起来呢？
因为c 在b的后面，所以让b从后往前遍历，然后b每往前移动一位，c就多了一个选择。此时，让d从最后开始往前遍历，与新加入的c做差存到hashmap中
题解
class Solution:
    def countQuadruplets(self, nums: List[int]) -> int:
        hashMap = collections.defaultdict(int)

        res = 0
        for b in range(len(nums)-2, 0, -1):
            # 将所有的 nums[d] - nums[c] 存到hashmap中
            for d in range(b+2, len(nums), 1):
                hashMap[nums[d]-nums[b+1]] += 1
            for a in range(0, b, 1):
                res += hashMap[nums[a]+nums[b]]

        return res