【每日一题】找出不同元素数目差数组

Tag

【哈希集合】【数组】【2024-01-31】

---

题目来源

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

---

解题思路

方法一：哈希集合

思路

使用哈希集合来存放数组中的元素，哈希集合可以实现去重功能。

在实现中，我们可以先计算后「后缀数组」，然后在更新前缀数组的同时更新答案数组。

算法

class Solution {
public:
    vector<int> distinctDifferenceArray(vector<int>& nums) {
        int n = nums.size(), suf[n+1];
        suf[n] = 0;
        unordered_set<int> s;
        for (int i = n - 1; i >= 0; --i) {
            s.insert(nums[i]);
            suf[i] = s.size();
        }

        s.clear();
        vector<int> res(n);
        for (int i = 0; i < n; ++i) {
            s.insert(nums[i]);
            res[i] = s.size() - suf[i+1];
        }
        return res;
    }
};

复杂度分析

时间复杂度：$O(n)$，$n$ 为数组 nums 的长度。

空间复杂度：$O(n)$。

---

写在最后

如果您发现文章有任何错误或者对文章有任何疑问，欢迎私信博主或者在评论区指出 💬💬💬。

如果大家有更优的时间、空间复杂度的方法，欢迎评论区交流。

最后，感谢您的阅读，如果有所收获的话可以给我点一个 👍 哦。