LeetCode 2670.找出不同元素数目差数组

给你一个下标从 0 开始的数组 nums ,数组长度为 n 。

nums 的 不同元素数目差 数组可以用一个长度为 n 的数组 diff 表示,其中 diff[i] 等于前缀 nums[0, ..., i] 中不同元素的数目 减去 后缀 nums[i + 1, ..., n - 1] 中不同元素的数目。

返回 nums 的 不同元素数目差 数组。

注意 nums[i, ..., j] 表示 nums 的一个从下标 i 开始到下标 j 结束的子数组(包含下标 i 和 j 对应元素)。特别需要说明的是,如果 i > j ,则 nums[i, ..., j] 表示一个空子数组。

示例 1:

输入:nums = [1,2,3,4,5]

输出:[-3,-1,1,3,5]

解释:

对于 i = 0,前缀中有 1 个不同的元素,而在后缀中有 4 个不同的元素。因此,diff[0] = 1 - 4 = -3 。

对于 i = 1,前缀中有 2 个不同的元素,而在后缀中有 3 个不同的元素。因此,diff[1] = 2 - 3 = -1 。

对于 i = 2,前缀中有 3 个不同的元素,而在后缀中有 2 个不同的元素。因此,diff[2] = 3 - 2 = 1 。

对于 i = 3,前缀中有 4 个不同的元素,而在后缀中有 1 个不同的元素。因此,diff[3] = 4 - 1 = 3 。

对于 i = 4,前缀中有 5 个不同的元素,而在后缀中有 0 个不同的元素。因此,diff[4] = 5 - 0 = 5 。

示例 2:

输入:nums = [3,2,3,4,2]

输出:[-2,-1,0,2,3]

解释:

对于 i = 0,前缀中有 1 个不同的元素,而在后缀中有 3 个不同的元素。因此,diff[0] = 1 - 3 = -2 。

对于 i = 1,前缀中有 2 个不同的元素,而在后缀中有 3 个不同的元素。因此,diff[1] = 2 - 3 = -1 。

对于 i = 2,前缀中有 2 个不同的元素,而在后缀中有 2 个不同的元素。因此,diff[2] = 2 - 2 = 0 。

对于 i = 3,前缀中有 3 个不同的元素,而在后缀中有 1 个不同的元素。因此,diff[3] = 3 - 1 = 2 。

对于 i = 4,前缀中有 3 个不同的元素,而在后缀中有 0 个不同的元素。因此,diff[4] = 3 - 0 = 3 。

提示:

1 <= n == nums.length <= 50

1 <= nums[i] <= 50

法一:处理前后缀不同元素数目数组:

cpp 复制代码
class Solution {
public:
    vector<int> distinctDifferenceArray(vector<int>& nums) {
        unordered_set<int> set;
        vector<int> tailDiff(nums.size());
        tailDiff[nums.size() - 1] = 0;
        // 此处要从倒数第二个开始遍历,因为只有一个元素时
        // set.insert(nums[i + 1]);会索引超出
        for (int i = nums.size() - 2; i >= 0; --i)
        {
            if (set.find(nums[i + 1]) == set.end())
            {
                tailDiff[i] = tailDiff[i + 1] + 1;
                set.insert(nums[i + 1]);
            }
            else
            {
                tailDiff[i] = tailDiff[i + 1];
            }
        }

        set.clear();
        vector<int> ans;
        vector<int> preDiff(nums.size());
        for (int i = 0; i < nums.size(); ++i)
        {
            if (set.find(nums[i]) == set.end())
            {
                if (i == 0)
                {
                    preDiff[i] = 1;
                }
                else
                {
                    preDiff[i] = preDiff[i - 1] + 1;
                }
                set.insert(nums[i]);
            }
            else
            {
                preDiff[i] = preDiff[i - 1];
            }

            ans.push_back(preDiff[i] - tailDiff[i]);
        }

        return ans;
    }
};

此算法时间复杂度为O(n),空间复杂度为O(n)。

法二:法一中,前缀数组其实是不需要的,因为第二个for循环中每遍历到一个元素就往set中加,set的大小就是前缀数组的值:

cpp 复制代码
class Solution {
public:
    vector<int> distinctDifferenceArray(vector<int>& nums) {
        unordered_set<int> set;
        vector<int> tailDiff(nums.size());
        tailDiff[nums.size() - 1] = 0;
        // 此处要从倒数第二个开始遍历,因为只有一个元素时
        // set.insert(nums[i + 1]);会索引超出
        for (int i = nums.size() - 2; i >= 0; --i)
        {
            if (set.find(nums[i + 1]) == set.end())
            {
                tailDiff[i] = tailDiff[i + 1] + 1;
                set.insert(nums[i + 1]);
            }
            else
            {
                tailDiff[i] = tailDiff[i + 1];
            }
        }

        set.clear();
        vector<int> ans;
        for (int i = 0; i < nums.size(); ++i)
        {
            set.insert(nums[i]);
            ans.push_back(set.size() - tailDiff[i]);
        }

        return ans;
    }
};

此算法时间复杂度为O(n),空间复杂度为O(n)。

相关推荐
CoovallyAIHub8 小时前
中科大DSAI Lab团队多篇论文入选ICCV 2025,推动三维视觉与泛化感知技术突破
深度学习·算法·计算机视觉
NAGNIP9 小时前
Serverless 架构下的大模型框架落地实践
算法·架构
moonlifesudo9 小时前
半开区间和开区间的两个二分模版
算法
moonlifesudo9 小时前
300:最长递增子序列
算法
CoovallyAIHub14 小时前
港大&字节重磅发布DanceGRPO:突破视觉生成RLHF瓶颈,多项任务性能提升超180%!
深度学习·算法·计算机视觉
CoovallyAIHub15 小时前
英伟达ViPE重磅发布!解决3D感知难题,SLAM+深度学习完美融合(附带数据集下载地址)
深度学习·算法·计算机视觉
聚客AI1 天前
🙋‍♀️Transformer训练与推理全流程:从输入处理到输出生成
人工智能·算法·llm
大怪v1 天前
前端:人工智能?我也会啊!来个花活,😎😎😎“自动驾驶”整起!
前端·javascript·算法
惯导马工2 天前
【论文导读】ORB-SLAM3:An Accurate Open-Source Library for Visual, Visual-Inertial and
深度学习·算法
骑自行车的码农2 天前
【React用到的一些算法】游标和栈
算法·react.js