Leetcode3002. 移除后集合的最多元素数

Every day a Leetcode

题目来源：3002. 移除后集合的最多元素数

解法1：贪心

可以将数组去重后分为三个部分：nums1 独有的，nums2 独有的，nums1 与 nums2 共有的。

求集合 S 时：

1. 先选择两个数组独有的。
2. 对于共有的，两个数组尽量选不一样的。

代码：

/*
 * @lc app=leetcode.cn id=10037 lang=cpp
 *
 * [10037] 移除后集合的最多元素数
 */

// @lc code=start
class Solution
{
public:
    int maximumSetSize(vector<int> &nums1, vector<int> &nums2)
    {
        int n = nums1.size();
        unordered_set<int> set1, set2;
        for (int &x : nums1)
            set1.insert(x);
        for (int &x : nums2)
            set2.insert(x);
        int common = 0; // 两个数组共有的元素个数
        for (int x : set1)
            if (set2.count(x))
                common++;
        // count1 和 count2 分别为数组 nums1 和 nums2 独有元素的个数
        int count1 = set1.size() - common, count2 = set2.size() - common;
        // 贪心策略：先选二者独有的，没得选才选二者共有的
        int s1 = min(count1, n / 2), s2 = min(count2, n / 2);
        return s1 + s2 + min(n - s1 - s2, common);
    }
};
// @lc code=end

结果：

复杂度分析：

时间复杂度：O(n)，其中 n 是数组 nums1/nums2 的长度。

空间复杂度：O(n)，其中 n 是数组 nums1/nums2 的长度。