PTA：7-3 两个递增链表的差集

题目

输入样例

5
1 3 5 7 9
3
2 3 5
1
2
3
4

输出样例

3
1 7 9
1
2

代码

#include 
#include 
#include 
using namespace std;
int main() 
{
    int n1, n2;
    cin >> n1;

    list<int> a;//①
    for (int i = 0; i < n1; i++) 
    {
        int value;
        cin >> value;
        a.push_back(value);
    }

    cin >> n2;//②
    list<int> b;
    for (int i = 0; i < n2; i++) 
    {
        int value;
       cin >> value;
        b.push_back(value);
    }

    unordered_set<int> b_set(b.begin(), b.end());//③
    list<int> result;

    for (const int& val : a) //④
    {
        if (b_set.find(val) == b_set.end()) 
        {
            result.push_back(val);
        }
    }

    cout << result.size() <<endl;//⑤

    for (auto it = result.begin(); it != result.end(); ++it) 
    {
        cout << *it;
        if (next(it) != result.end()) 
        {//⑦
            cout << " ";
        }
    }
    return 0;
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

①输入第一个链表的元素个数、元素

②输入第二个链表的元素个数、元素
③用unordered_set来存储第二个列表的元素，以便快速查找

④遍历第一个列表，如果元素在第二个列表中找不到，则加入结果列表

⑤输出结果列表的size

⑥最后输出结果列表的元素，用空格分隔

注意末尾没有空格，需要加上这个判断