2024.1.31每日一题

LeetCode

找出不同元素数目差数组

2670. 找出不同元素数目差数组 - 力扣（LeetCode）

题目描述

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

思路

该思路来自灵神

哈希表

先倒序枚举 nums，并记录每个后缀的不同元素个数到suf数组中。

正序枚举 num[i]，同样地，记录每个前缀的不同元素个数，减去 suf[i+1]，即为答案。

代码

C++

class Solution {
public:vector<int> distinctDifferenceArray(vector<int>& nums) {int n = nums.size();int suf[n + 1];vector<int> ans(n);suf[n] = 0;unordered_set<int> s;for(int i = n - 1; i > 0 ;i--){s.insert(nums[i]);suf[i] = s.size();}s.clear();for(int i = 0; i < n; i++){s.insert(nums[i]);ans[i] = s.size() - suf[i + 1];}return ans;}
};

Java

class Solution {public int[] distinctDifferenceArray(int[] nums) {int n = nums.length;int[] suf = new int[n+1];HashSet<Integer> s = new HashSet<>(); int[] ans = new int[n]; // 最终结果for(int i = n - 1; i > 0;i--){ // 先倒序遍历s.add(nums[i]);suf[i] = s.size();}s.clear();for(int i = 0;i < n;i++){s.add(nums[i]);ans[i] = s.size() - suf[i + 1];}return ans;}
}

今日所学

Java中的HashSet是一种无序、不可重复的集合，它是通过哈希表实现的。

add()方法的定义为：

public boolean add(E e)

其中，E是泛型类型，表示元素的类型。add()方法的作用是将指定的元素添加到集合中。如果添加成功，则返回true；如果该元素已经存在于集合中，则返回false。

使用示例

HashSet<Integer> set = new HashSet<>();
set.add(1); // 添加一个整数1
set.add(2); // 添加一个整数2
set.add(3); // 添加一个整数3
set.add(1); // 再次添加整数1（已存在于集合中）
System.out.println(set); // 输出：[1, 2, 3]