LeetCode 2670. 找出不同元素数目差数组，哈希计数+前后缀分离

一、题目

1、题目描述

给你一个下标从 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] 表示一个空子数组。

2、接口描述

​

class Solution {
public:
    vector<int> distinctDifferenceArray(vector<int>& nums) {
        
    }
};

3、原题链接

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

---

二、解题报告

1、思路分析

正序遍历一次，累计前缀，逆序遍历一次，减去后缀

2、复杂度

时间复杂度： O(n) 空间复杂度：O(n)

3、代码详解

​

class Solution {
public:
    vector<int> distinctDifferenceArray(vector<int>& nums) {
        vector<int> ret;
        unordered_set<int> mp;
        for(auto x : nums) mp.insert(x), ret.emplace_back(mp.size());
        mp.clear();
        for(int i = nums.size() - 1; i >= 0; i--) ret[i] -= mp.size(), mp.insert(nums[i]);
        return ret;
    }
};