C++ 集合的基本运算 | 使用STL的set_difference set_intersection set_union实现集合和交集 并集 差集

八辈子没写C++的人今天被这个题逼上来了。

oj

输入格式：

输入两个集合

输出格式：

输出两个集合元素，并输出他们的并集、交集和差集。

输入样例1：

3 4 5 6 8 0
1 2 3 7 9 10 0

输出样例1：

A = { 3, 4, 5, 6, 8 }
B = { 1, 2, 3, 7, 9, 10 }
C = A+B = { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 }
C = A*B = { 3 }
C = A-B = { 4, 5, 6, 8 }

提示：

有一点需要注意的是，给你的样例是有序的，输出也是有序的。给你样例无序，你的输出还要是有序的。

答案：

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
void print(vector<int> arr);
vector<int> input(vector<int> arr);
int main()
{
    vector<int>a,b,res;
    a = input(a);
    b = input(b);
    sort(a.begin(),a.end());
    sort(b.begin(),b.end());

    cout << "A";
    print(a);

    cout << "B";
    print(b);

    set_union(a.begin(), a.end(),b.begin(), b.end(),std::back_inserter(res));
    cout << "C = A+B";
    print(res);
    res.clear();
    
    set_intersection(a.begin(), a.end(),b.begin(), b.end(),std::back_inserter(res));
    cout << "C = A*B";
    print(res);
    res.clear();

    set_difference(a.begin(), a.end(),b.begin(), b.end(),std::back_inserter(res));
    cout << "C = A-B";
    print(res);
    res.clear();
    
    return 0;
}
void print(vector<int> arr)
{
    cout << " = { ";
    for (auto it = arr.begin(); it != arr.end(); it++)
    {
        if (it != arr.end() - 1)
            cout << *it << ", ";
        else
            cout << *it;

    }
    cout << " }\n";
}
vector<int> input(vector<int> arr)
{
    int n;
    while (cin >> n)
    {
        if (n != 0)
            arr.push_back(n);
        else
            break;
    }
    return arr;
}

代码解析：

上边用到了两个函数，先声明了，函数体放在main函数后边。别的没什么课讲的，就是按题目操作。

STL讲解

上边这个题，得亏用库函数写了。我本来用python写的，写完以后发现代码只能用C++提交，内心崩溃。

那就讲一下这三个库函数怎么用。先放一段完整代码，解析看底下。

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
void print(vector<int> arr);
int main()
{
    vector<int>result;
    vector<int> a{ 1, 4, 2, 5, 8 };
    vector<int> b{ 9, 4, 2, 7, 0 };

    sort(a.begin(),a.end());
    sort(b.begin(),b.end());

    cout << "A";
    print(a);

    cout << "B";
    print(b);

    set_union(a.begin(), a.end(),b.begin(), b.end(),std::back_inserter(result));
    cout << "A+B";
    print(result);
    result.clear();
    
    set_intersection(a.begin(), a.end(),b.begin(), b.end(),std::back_inserter(result));
    cout << "A*B";
    print(result);
    result.clear();

    set_difference(a.begin(), a.end(),b.begin(), b.end(),std::back_inserter(result));
    cout << "A-B";
    print(result);
    result.clear();
    
    return 0;
}
void print(vector<int> arr)
{
    cout << " = { ";
    for (auto it = arr.begin(); it != arr.end(); it++)
        cout << *it<<" ";
    cout << "}\n";
}

• 首先要加上头文件：

  #include<algorithm>
  

• 需要两个待操作的集合和一个存储结果的集合。但是注意这里用的并不是set，而是vector。
  所以声明三个集合

  vector<int> a,b,result;
  

• 交叉并集直接调用库函数：

  	# 并集
  	set_union(a.begin(), a.end(),b.begin(), b.end(),std::back_inserter(result));
  	# 交集
      set_intersection(a.begin(), a.end(),b.begin(), b.end(),std::back_inserter(result));
  	# 差集
      set_difference(a.begin(), a.end(),b.begin(), b.end(),std::back_inserter(result));
  

  这里需要注意的是：

  • a，b必须是有序的，如果传入是无序的需要自己加上一步排序，所以我在代码中加上了sort(a.begin(),a.end());
  • 执行结果是back_inserter，在尾部插入，所以要么声明多个变量存储结果，要么在使用完后注意清空result.clear();，否则结果会叠加。比如下边这样：