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

83 阅读3分钟

题目要求:

给你一个下标从 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

链接leetcode.cn/problems/fi…

解题思路

使用集合的方式计算不同元素个数,需要注意的是要遍历两次,第一次是顺序计算当前值不同的元素个数,第二次需要降序遍历,计算当前值顺序不同元素和降序不同元素的差值。这里有个细节,第二次降序遍历的时候因为取值是左闭右开,所以需要取第一个值的时候,不能先写集合,应该先计算。

代码

func distinctDifferenceArray(nums []int) []int {
	// 考虑临界情况1个元素的情况
	if len(nums) == 1 {
		return []int{1}
	}
	temp := make([]int, 0)
	st := make(map[int]struct{})
	for i := 0; i < len(nums); i++ {
		st[nums[i]] = struct{}{}
		temp = append(temp, len(st))
	}
	st = map[int]struct{}{}
	for i := len(nums) - 1; i >= 0; i-- {
		temp[i] -= len(st)
                st[nums[i]] = struct{}{}
	}
	return temp
}