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

输入格式

一个数组nums,表示下标从 0 开始的原始数组。

输出格式

一个数组diff,表示不同元素数目差数组。

数据范围

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

1 <= nums[i] <= 50

输入样例:

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 。

题目分析

Hash Table, Prefix Sum

代码实现

const int N = 100;
class Solution {
public:
    int pre[N], suf[N], n;
    unordered_set <int> st[2];
    vector<int> distinctDifferenceArray(vector<int>& nums) {
        n = nums.size();
        for(int i = 0; i < n; ++i) {
            st[0].insert(nums[i]);
            pre[i] = st[0].size();
            st[1].insert(nums[n - i - 1]);
            suf[n - i - 1] = st[1].size();
        }
        vector <int> dif(n);
        for(int i = 0; i < n; ++i) {
            dif[i] = pre[i] - suf[i + 1];
        }
        return dif;
    }
};
分类: HashPrefixSum

0 条评论

发表回复

Avatar placeholder

您的电子邮箱地址不会被公开。 必填项已用*标注

友情链接:Ctips' blog, Colza’s blog

站点状态:Status