分类讨论 -- Choose the Different ones!-CSDN博客

本文链接：https://blog.csdn.net/killer_Li_yy/article/details/136406970

前言：

时常感觉自己学的算法，做的题目，过不了多久就忘记了，故想用博客记录自己学习过的算法和遇见的好题目。

这是题目源于codeforces

题目大意：

给定一个长度为 n 的数组 {a}、一个长度为 m 的数组 {b} 和一个偶数 k，求是否能在 a 和 b 中各选 k/2 个数，使得这些数包括从 1 到 k 的所有整数。

主要记录一下重要思路，以及代码实现，具体的题目可以在codeforces 或者洛谷中找到。

原题链接：Choose the Different Ones!

思路：

首先大于数组a、b中大于K的数一定不符合题意。

在排除掉大于K的数据后，我们可以将数值分为三类：

1. 只存在数组a

2. 只存在数组b

3. 同时出现在两个数组中

本题如果顺着题意模拟的话，需要考虑的细节会比较多，代码写起来不好下手。所以我们采取正难则反的思路。排除掉不符合的情况。

如果满足以下任一条件，则答案为 "NO"：

第一种类型的值的数量大于 k/2 (这意味着我们无法选择所有此类元素)；
第二种类型的值的数量大于 k/2 (这意味着我们无法选择所有此类元素)；
所有三种类型的值的总数小于 k (这意味着有些值没有被选中。(这意味着某些值不出现在任一数组中)。

否则答案就 "YES"

实现代码：

#include <iostream>
#include <vector>
 
using namespace std;
 
void slove() {
    int m, n, k;
    cin >> n >> m >> k;
    vector<int> a(n); vector<int> b(m);
    for (int i = 0; i < n; i++) {
        cin >> a[i];
    }
    for (int i = 0; i < m; i++) {
        cin >> b[i];
    }
 
    vector<int> cnt(k + 1, 0);
    // 记录只在a出现的元素
    for (auto e : a) {
        if (e <= k) {
            cnt[e] |= 1;
        }
    }
    // 记录只在b出现的元素
    for (auto e : b) {
        if (e <= k) {
            cnt[e] |= 2;
        }
    }
    
    // 确定三种情况
    vector<int> c(4, 0);
    //  cnt取值为0， 1， 2， 3，分别表示否定的三种情况
    for (auto e : cnt) {
        c[e]++;
    }
 
    if (c[1] > k / 2 || c[2] > k / 2 || c[1] + c[2] + c[3] != k) {
        cout << "NO" << endl;
    } else {
        cout << "YES" << endl;
    }
}
 
int main() {
    int t; cin >> t;
    while (t--) {
        slove();
    }
    return 0;
}

感谢你的阅读！欢迎评论区讨论！