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

---

解题分析及思路：

为了解决这个问题，可以使用哈希表来记录前缀和后缀中每个元素的出现次数，并利用这些信息来计算差值。

首先使用两个哈希表 preM 和 sufM 来记录前缀和后缀中每个元素的出现次数。同时，使用两个变量 preCount 和 sufCount 分别记录前缀和后缀中不同元素的数目。

接下来，遍历数组 nums，在每个位置更新前缀和后缀的哈希表以及相应的计数变量。最后，计算差值并将结果存储在数组 result 中。

func distinctDifferenceArray(nums []int) []int {
	preM := make(map[int]int)
	sufM := make(map[int]int)
	preCount := 0
	sufCount := 0
	// 预处理后缀
	for index := range nums {
		if count, ok := sufM[nums[index]]; ok {
			sufM[nums[index]] = count + 1
		} else {
			sufM[nums[index]] = 1
			sufCount++
		}
	}
	result := make([]int, len(nums))
	for index := range nums {
		// 处理前缀
		if count, ok := preM[nums[index]]; ok {
			preM[nums[index]] = count + 1
		} else {
			preM[nums[index]] = 1
			preCount++
		}
		// 处理后缀
		sufM[nums[index]] = sufM[nums[index]] - 1
		if sufM[nums[index]] == 0 {
			sufCount--
		}
		// 计算结果
		result[index] = preCount - sufCount
	}
	return result
}

复杂度：

• 时间复杂度：O(N)，其中 n 为输入数组 nums 的长度。
• 空间复杂度：O(N)，其中 n 为输入数组 nums 的长度。

执行结果：

• 执行耗时:22 ms,击败了26.09% 的Go用户
• 内存消耗:5.5 MB,击败了56.52% 的Go用户