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

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

由于答案可能非常大，请返回结果对 10⁹ + 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 <= 10⁵
0 <= nums[i] <= 10⁵

解题方法：哈希表计数

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

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

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

注意：

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

时空复杂度分析

时间复杂度 O ( l e n ( n u m s ) ) O(len(nums)) O(len(nums))
空间复杂度 O ( l e n ( n u m s ) ) O(len(nums)) O(len(nums))

AC代码

C++

cpp 复制代码

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

千篇源码题解已开源