1822 차집합

처음에 count 배열 만들어서 a 원소 전부 1로 만들고 b원소 전부 0으로 만든 다음 남아있는 1을 출력하려고 했다.

 

하지만 이러면 n이 50만이라 메모리가 터진다. 

 

반복문 두 번 써서 확인하는 방법도 있지만 이러면 시간이 터짐

 

 

그래서 정렬한 다음 비교하는 방법을 떠올렸다. 

 

#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
#include <functional>
#include <string>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <cmath>
#include <cstring>
#include <bitset>

#define xx first
#define yy second
#define all(x) (x).begin(), (x).end()
#define MAXV 1000000

using namespace std;
using i64 = long long int;
using ii = pair<int, int>;
using ii64 = pair<i64, i64>;
using iii = tuple<int, int, int>;


int     main()
{
    int n, m;
    scanf("%d %d", &n, &m);

    vector<int> va(n);
    vector<int> vb(n);
    for (int i = 0; i < n; i++)
        scanf("%d", &va[i]);
    for (int i = 0; i < m; i++)
        scanf("%d", &vb[i]);

    sort(all(va));
    sort(all(vb));

    vector<int> ans;
    int i = 0, j = 0;
    while (i < n && j < m)
    {
        if (va[i] < vb[j])
        {
            ans.push_back(va[i]);
            i++;
        }
        else if (va[i] > vb[j])
            j++;
        else
            i++, j++;
    }

    for ( ;i < n; i++)
    {
        ans.push_back(va[i]);
    }

    printf("%d\n", ans.size());
    for (int i = 0; i < ans.size(); i++)
        printf("%d ", ans[i]);

    return 0;
}

 

하지만 틀렸군요 문제가 무엇일까요?

 

저번이랑 비슷한 문제였다ㅋㅋㅋㅋ

vector<int> vb(n);

m이라 했어야 했다

 

#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
#include <functional>
#include <string>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <cmath>
#include <cstring>
#include <bitset>

#define xx first
#define yy second
#define all(x) (x).begin(), (x).end()
#define MAXV 1000000

using namespace std;
using i64 = long long int;
using ii = pair<int, int>;
using ii64 = pair<i64, i64>;
using iii = tuple<int, int, int>;


int     main()
{
    int n, m;
    scanf("%d %d", &n, &m);

    vector<int> va(n);
    vector<int> vb(m);
    for (int i = 0; i < n; i++)
        scanf("%d", &va[i]);
    for (int i = 0; i < m; i++)
        scanf("%d", &vb[i]);

    sort(all(va));
    sort(all(vb));

    vector<int> ans;
    int i = 0, j = 0;
    while (i < n && j < m)
    {
        if (va[i] < vb[j])
        {
            ans.push_back(va[i]);
            i++;
        }
        else if (va[i] > vb[j])
            j++;
        else
            i++, j++;
    }

    for ( ;i < n; i++)
    {
        ans.push_back(va[i]);
    }

    printf("%d\n", ans.size());
    for (int i = 0; i < ans.size(); i++)
        printf("%d ", ans[i]);

    return 0;
}