详细介绍：LeetCode 3583.统计特殊三元组：哈希表计数

【LetMeFly】3583.统计特殊三元组：哈希表计数

力扣题目链接：https://leetcode.cn/problems/count-special-triplets/

给你一个整数数组 nums。

特殊三元组 定义为满足以下条件的下标三元组 (i, j, k)：

• 0 <= i < j < k < n，其中 n = nums.length
• nums[i] == nums[j] * 2
• nums[k] == nums[j] * 2

返回数组中 特殊三元组 的总数。

由于答案可能非常大，请返回结果对 109 + 7 取余数后的值。

示例 1：

输入：nums = [6,3,6]

输出：1

解释：

唯一的特殊三元组是 (i, j, k) = (0, 1, 2)，其中：

• nums[0] = 6, nums[1] = 3, nums[2] = 6
• nums[0] = nums[1] * 2 = 3 * 2 = 6
• nums[2] = nums[1] * 2 = 3 * 2 = 6

示例 2：

输入：nums = [0,1,0,0]

输出：1

解释：

唯一的特殊三元组是 (i, j, k) = (0, 2, 3)，其中：

• nums[0] = 0, nums[2] = 0, nums[3] = 0
• nums[0] = nums[2] * 2 = 0 * 2 = 0
• nums[3] = nums[2] * 2 = 0 * 2 = 0

示例 3：

输入：nums = [8,4,2,8,4]

输出：2

解释：

共有两个特殊三元组：

• (i, j, k) = (0, 1, 3)

  
  	
  nums[0] = 8, nums[1] = 4, nums[3] = 8
  	
  nums[0] = nums[1] * 2 = 4 * 2 = 8
  	
  nums[3] = nums[1] * 2 = 4 * 2 = 8
  
  

• (i, j, k) = (1, 2, 4)
  • nums[1] = 4, nums[2] = 2, nums[4] = 4
  • nums[1] = nums[2] * 2 = 2 * 2 = 4
  • nums[4] = nums[2] * 2 = 2 * 2 = 4

提示：

• 3 <= n == nums.length <= 105
• 0 <= nums[i] <= 105

解题方法：哈希表计数

使用一个哈希表，统计nums中每个数分别出现了多少次。

接着再次遍历nums数组，再使用一个哈希表，统计到目前为止每个数分别出现了多少次。

假设当前遍历元素为$t$，那么在此之前$2t$出现的次数乘以$2t$在此之后出现的次数($出现总次数-此前出现次数$)累加到答案中。

注意：

1. 特判$t=0$时
2. 其实也可以使用64位整数并且只在最后取一次模

时空复杂度分析

• 时间复杂度$O(len(nums))$
• 空间复杂度$O(len(nums))$

AC代码

C++

/*
* @LastEditTime: 2025-12-09 18:43:53
*/
typedef long long ll;
const long long MOD = 1e9 + 7;
class Solution {
public:
int specialTriplets(vector<int>& nums) {
  ll ans = 0;
  unordered_map<int, int> total, now;
    for (int t : nums) {
    total[t]++;
    }
    for (int t : nums) {
    int finding = t * 2;
    if (now.count(finding)) {
    ll left = now[finding];
    ans = (ans + left * (total[finding] - left - (t == 0))) % MOD;
    }
    now[t]++;
    }
    return ans;
    }
    };

同步发文于CSDN和我的个人博客，原创不易，转载经作者同意后请附上原文链接哦~

千篇源码题解已开源