蓝桥-ALGO-10-集合运算

ACM模版

描述

题解

这个题如果全部用 set 写，那么就很简单了，如果用数组写，那就是一个细心问题，别粗心害死自己。

我用的方法效率挺可观的，竟然超时了，后来发现，我开的数组不够大，因为每个数组都是不超过 1000 个元素，那么并起来就是不超过两千个，我只开了一千余个肯定不够用~~~

代码

#include <iostream>
#include <algorithm>
#include <set>

using namespace std;

const int MAXN = 2222;
const int INF = 0x7fffffff;

int A[MAXN], B[MAXN];
int C[MAXN], D[MAXN];
set<int> E;

int main(int argc, const char * argv[])
{
//    freopen("/Users/zyj/Desktop/input.txt", "r", stdin);
    int n, m;
    cin >> n;
    for (int i = 0; i < n; i++)
    {
        cin >> A[i];
    }
    cin >> m;
    for (int i = 0; i < m; i++)
    {
        cin >> B[i];
    }
    sort(A, A + n);
    sort(B, B + m);
    A[n] = B[m] = INF;

    int i = 0, j = 0;
    int cntC = 0, cntD = 0;
    if (A[0] < B[0])
    {
        while (i < n)
        {
            D[cntD++] = A[i];
            while (j < m && B[j] < A[i + 1])
            {
                if (B[j] != A[i])
                {
                    D[cntD++] = B[j];
                }
                else
                {
                    C[cntC++] = B[j];
                }
                j++;
            }
            i++;
        }
        while (j < m)
        {
            D[cntD++] = B[j];
            j++;
        }
    }
    else
    {
        while (j < m)
        {
            D[cntD++] = B[j];
            while (i < n && A[i] < B[j + 1])
            {
                if (B[j] != A[i])
                {
                    D[cntD++] = A[i];
                }
                else
                {
                    C[cntC++] = A[i];
                }
                i++;
            }
            j++;
        }
        while (i < n)
        {
            D[cntD++] = A[i];
            i++;
        }
    }

    for (int i = 0; i < cntC; i++)
    {
        cout << C[i] << ' ';
        E.insert(C[i]);
    }
    cout << '\n';
    for (int i = 0; i < cntD; i++)
    {
        cout << D[i] << ' ';
    }
    cout << '\n';
    for (int i = 0; i < n; i++)
    {
        if (!E.count(A[i]))
        {
            cout << A[i] << ' ';
        }
    }
    cout << '\n';

    return 0;
}