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

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

---

提示 1

Which data structure will help you maintain distinct elements?

---

提示 2

Iterate over all possible prefix sizes. Then, use a nested loop to add the elements of the prefix to a set, and another nested loop to add the elements of the suffix to another set.

---

解法：一维前缀和

class Solution {
    public int[] distinctDifferenceArray(int[] nums) {
        int n = nums.length;
        int[] diff = new int[n];
        Set<Integer> prefixSet = new HashSet<Integer>();
        Set<Integer> suffixSet = new HashSet<Integer>();
        int[] prefix = new int[n];
        int[] suffix = new int[n + 1];
        for (int i = 0; i < n; i++) {
            prefixSet.add(nums[i]);
            suffixSet.add(nums[n - 1 - i]);
            prefix[i] = prefixSet.size();
            suffix[n - 1 - i] = suffixSet.size();
        }
        suffix[n] = 0;
        for (int i = 0; i < n; i++) {
            // prefix[i]: nums[0..i]种类数
            // suffix[i]: nums[i..n - 1]种类数
            diff[i] = prefix[i] - suffix[i + 1];
        }
        return diff;
    }
}

复杂度分析 

• 时间复杂度：O(n)，n 是 数组 nums 的长度。总的时间复杂度 O(2n) = O(n)。
• 空间复杂度：O(n)，n 是 数组 nums 的长度。

同类题型：

​​​​​​​LeetCode 724. 寻找数组的中心下标-CSDN博客

​​​​​​​LeetCode 1991. 找到数组的中间位置-CSDN博客LeetCode 2574. 左右元素和的差值-CSDN博客

LeetCode 2574. 左右元素和的差值-CSDN博客

LeetCode 2219. 数组的最大总分-CSDN博客