题目:
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
解答:
分别用两个集合和两个数组来记录前后缀不同元素的个数,当集合中不包含该元素时,将该元素加入集合中并用数组记录此时集合的大小,即为不同元素的个数。
代码:
class Solution {
    public int[] distinctDifferenceArray(int[] nums) {
        int n=nums.length;
        Set<Integer> set1=new HashSet<>();
        Set<Integer> set2=new HashSet<>();
        int[] prefix=new int[n];
        int[] suffix=new int[n];
        int[] diff=new int[n];
        for(int i=0;i<n;i++){
            if(!set1.contains(nums[i])){
                set1.add(nums[i]);
            }
            prefix[i]=set1.size();
        } 
        for(int i=n-1;i>=1;i--){
            if(!set2.contains(nums[i])){
                set2.add(nums[i]);
            }
            suffix[i-1]=set2.size();
        }
        suffix[n-1]=0;
        for(int i=0;i<n;i++){
            diff[i]=prefix[i]-suffix[i];
        }
        return diff;
    }
}
结果:












![XCTF:warmup[WriteUP]](https://img-blog.csdnimg.cn/direct/33a1b03297744e6db6fddc3d3e739300.png)







