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)

  <ul>
  	<li><code>nums[0] = 8</code>, <code>nums[1] = 4</code>, <code>nums[3] = 8</code></li>
  	<li><code>nums[0] = nums[1] * 2 = 4 * 2 = 8</code></li>
  	<li><code>nums[3] = nums[1] * 2 = 4 * 2 = 8</code></li>
  </ul>
  </li>
  <li><code>(i, j, k) = (1, 2, 4)</code>
  <ul>
  	<li><code>nums[1] = 4</code>, <code>nums[2] = 2</code>, <code>nums[4] = 4</code></li>
  	<li><code>nums[1] = nums[2] * 2 = 2 * 2 = 4</code></li>
  	<li><code>nums[4] = nums[2] * 2 = 2 * 2 = 4</code></li>
  </ul>
  </li>
  

 

提示：

• 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和我的个人博客，原创不易，转载经作者同意后请附上原文链接哦~

千篇源码题解已开源