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到n-1，创建两个map，a和b，a存储0到i不同元素数目，b存储i+1到n-1的不同元素数目，然后size相减得到diff。

代码：

class Solution {
public:
    vector<int> distinctDifferenceArray(vector<int>& nums) {
        int n = nums.size();
        vector<int> diff(n, 0);
        map<int, int> a;
        map<int, int> b;
        int i = 0;
        for (i = 0; i < n; i++) {
            b[nums[i]]++;
        }
        for (i = 0; i < n; i++) {
            a[nums[i]]++;
            b[nums[i]]--;
            if (b[nums[i]] == 0)
                b.erase(nums[i]);
            diff[i] = a.size() - b.size();
        }
        return diff;
    }
};

复杂度分析：

该方法的时间复杂度为O( )，其中  是数组的长度，因为它包括两个 map 容器的插入和查询操作，时间复杂度为 O( )。空间复杂度也为 O(N)，因为额外使用了两个 map 容器来统计元素的个数。(来自gpt3.5，应为我不会算复杂度)

这个方法确实是解出来了，但是不是太优。接下来学习官方题解。

代码：

class Solution {
public:
    vector<int> distinctDifferenceArray(vector<int>& nums) {
        int n = nums.size();
        unordered_set<int> st;
        vector<int> sufCnt(n + 1, 0);
        for (int i = n - 1; i > 0; i--) {
            st.insert(nums[i]);
            sufCnt[i] = st.size();
        }
        vector<int> res;
        st.clear();
        for (int i = 0; i < n; i++) {
            st.insert(nums[i]);
            res.push_back(int(st.size()) - sufCnt[i + 1]);
        }
        return res;
    }
};

作者：力扣官方题解
链接：https://leetcode.cn/problems/find-the-distinct-difference-array/solutions/2614624/zhao-chu-bu-tong-yuan-su-shu-mu-chai-shu-iufa/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

首先学习到了一个我没见过的unordered_set类型：

C++常用语法——unordered_set_c++ unordered_set-CSDN博客

unordered_set 容器，哈希表，无序，元素唯一。

在这个题中用到的是st.size()，因为容器中元素的唯一性，所以可以根据这个来判断不同元素个数。

在第一个循环中我们将前i位的不同元素个数存储在sufCnt中，第二个循环，我们重新计算st.size()注意两个循环之间有一个st.clear()，用当前循环的st.size()也就是前i位的不同元素个数减去上一个循环已经存储好的sufCnt[i+1]就可以得到结果。

欢迎大家和我交流。