【力扣每日一题】力扣2670找出不同元素数目差数组

题目来源

力扣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] 表示一个空子数组。

解题思路

记录前缀和后缀的不同元素数目就行了。

代码实现

java实现

public class Solution {
    public int[] distinctDifferenceArray(int[] nums) {
		// 前缀
        Set<Integer> prefixSet = new HashSet<>();
		// 后缀
        Map<Integer, Integer> suffixMap = new HashMap<>();
        int[] answer = new int[nums.length];
        for (int i = 0; i < nums.length; i++) {
            suffixMap.put(nums[i], suffixMap.getOrDefault(nums[i], 0) + 1);
        }
        answer[nums.length - 1] = suffixMap.size();
        for (int i = 0; i < nums.length - 1; i++) {
            Integer rest = suffixMap.get(nums[i]);
            if (rest == 1) {
                suffixMap.remove(nums[i]);
            }else {
                suffixMap.put(nums[i], rest - 1);
            }
            prefixSet.add(nums[i]);
            answer[i] = prefixSet.size() - suffixMap.size();
        }
        return answer;
    }
}

c++实现

class Solution {

public:
    vector<int> distinctDifferenceArray(vector<int>& nums) {
        unordered_set<int> prefixSet = unordered_set<int>();
        unordered_map<int, int> suffixMap = unordered_map<int,int>();
        int count = 0;
        vector<int> answer= vector<int>(nums.size());
        for (int i = 0; i < nums.size(); i++) {
            suffixMap[nums[i]]++;
            count += (suffixMap[nums[i]] == 1);
        }
        answer[nums.size() - 1] = suffixMap.size();
        for (int i = 0; i < nums.size() - 1; i++) {
            int rest = suffixMap[nums[i]];
            if (rest == 1) {
                count--;
            }
            else {
                suffixMap[nums[i]]--;
            }
            prefixSet.insert(nums[i]);
            answer[i] = prefixSet.size() - count;
        }
        return answer;
    }
};