找出不同元素数目差数组

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.示例

输入：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

3.思路

  这题看着比较难理解，但根据示例的计算逻辑我们可以知道这题是让我们求一个给定数组nums的元素数目前后缀数组。如果暴力模拟的话我们可以用一个循环来遍历数组，然后我们需要借助一个特殊的数据结构来求元素数目个数——集合。集合set它是一个无序且无重复的数据结构。然后我们直接利用python的切片来获取数组nums的前后缀数组并转换为set类型，但这个并不是很好的方法，这里用到了大量的切片转换的操作，对空间复杂度和时间复杂度有很大的开销。

  那我们如何优化暴力模拟的解法呢？对于前缀数组元素数目和，我们可以定义一个全局变量集合，然后在遍历nums数组的时候往set里面添加元素并及时获取set的长度，这个长度就是当前前缀数组的元素数目和——前缀数组数目和。

同理，我们先清空刚才的set集合，然后我们可以反向遍历nums数组，每次往里面添加元素并及时获取set的长度，这个长度就是nums[i]到nums[n-1]之间的元素数目和——后缀数组元素数目和。

4.代码

length = len(nums)
        tmp = [0 for i in range(length)]    # 后缀元素数目和数组
        tmp.append(0)    # 最后一个元素的后缀数目为0
        rst = []    # 结果数组
        s = set()    # 集合
        # 反向遍历数组并更新后缀元素数目和数组
        for i in range(length-1,-1,-1):
            s.add(nums[i])
            tmp[i] = len(s)   # i 到 n-1之间不同元素数目

        s.clear() # 清除集合元素
        # 正向遍历数组获取前缀元素数目和
        for i in range(length):
            s.add(nums[i])
            front_num = len(s)  # 前缀数目
            behond_num = tmp[i+1]    # 后缀数目
            rst.append(front_num-behond_num)    # 添加到结果数组
        return rst