2025.12.09 力扣每日一题

小白程序员成长日记2025-12-11 11:19
简单做法溢出了，题目题解里的做法好！
cpp 复制代码
class Solution {
public:
    int specialTriplets(vector<int>& nums) {
        //定义取模常量
        const int MOD = 1000000007;
        //初始化后缀哈希表，统计【整个数组中】每个数值出现的次数
        //初始化时，suf包含所有元素，代表j还没开始遍历，所有元素都在j右侧
        unordered_map<int, int> suf;
        for (int x : nums) {
            suf[x]++;//遍历数组，每个数值x的计数+1
        }
        //初始化答案变量，long long防止溢出
        long long ans = 0;
        //初始化前缀哈希表pre：统计【j左侧（0~j-1】每个数值的出现次数，初始为空
        unordered_map<int, int> pre;
        //遍历数组，当前遍历的元素x就是nums[j](j为中间下标)
        for (int x : nums) {
            //关键：把当前x从suf中减1
            //因为当前x是nums[j]，j的位置已经确定，它不属于「j右侧的元素」了，所以要从后缀中移除
            suf[x]--;
            //核心计算：以当前j为中间点的三元组数量 = 左边符合条件的数量 × 右边符合条件的数量
            //pre[x*2]：j左侧（0~j-1）中值为x*2的元素个数（即满足nums[i]=2*nums[j]的i的数量）
            //suf[x*2]：j右侧（j+1~n-1）中值为x*2的元素个数（即满足nums[k]=2*nums[j]的k的数量）
            //LL：强制转为long long，避免int相乘溢出
            ans += 1LL * pre[x * 2] * suf[x * 2];
            // 把当前x加入前缀哈希表pre
            //因为遍历完当前j后，下一个j'会在当前j的右侧，当前j就属于「j'左侧的元素」了
            pre[x]++;
        }
        //返回取余结果
        return ans % MOD;
    }
};