18.6【STL常用集合算法set_intersection、set_union、set_difference】

#include<iostream>
#include<cstdlib>
using namespace std;
#include<vector>
#include<algorithm>


/*
    5.6 常用集合算法

    5.6.1 set_intersection

        求两个容器的交集
        set_intersection(iterator beg1, iterator end1, iterator beg2, iterator end2, iterator dest);
        // 求两个集合的交集
        // 注意:两个集合必须是有序序列
        // beg1 容器1开始迭代器 // end1 容器1结束迭代器 // beg2 容器2开始迭代器 // end2 容器2结束迭代器 //
        dest 目标容器开始迭代器

    5.6.2 set_union

        求两个集合的并集
        set_union(iterator beg1, iterator end1, iterator beg2, iterator end2, iterator dest);
        // 求两个集合的并集
        // 注意:两个集合必须是有序序列
        // beg1 容器1开始迭代器 // end1 容器1结束迭代器 // beg2 容器2开始迭代器 // end2 容器2结束迭代器 //
        dest 目标容器开始迭代器

    5.6.3 set_difference

        求两个集合的差集
        set_difference(iterator beg1, iterator end1, iterator beg2, iterator end2, iterator dest);
        // 求两个集合的差集
        // 注意:两个集合必须是有序序列
        // beg1 容器1开始迭代器 // end1 容器1结束迭代器 // beg2 容器2开始迭代器 // end2 容器2结束迭代器 //
        dest 目标容器开始迭代器
*/


void myprint(int val)
{
    cout << val << " ";
}


void test561()
{
    vector<int> v1;
    vector<int> v2;
    for(int i=0; i<10; i++)
    {
        v1.push_back(i);
        v2.push_back(i+5);
    }

    vector<int> v_tar;
    //极端情况：大容器包含小容器，开辟空间时取小容器size即可
    v_tar.resize(min(v1.size(), v2.size())); //min()需要头文件<algorithm>

    vector<int>::iterator pos_end = set_intersection(v1.begin(), v1.end(), v2.begin(), v2.end(), v_tar.begin()); //返回v_tar实际元素结束位置

    //for_each(v_tar.begin(), v_tar.end(), myprint); //全部元素：5 6 7 8 9 0 0 0 0 0
    for_each(v_tar.begin(), pos_end, myprint); //实际元素：5 6 7 8 9
    cout << endl;
}


void test562()
{
    vector<int> v1;
    vector<int> v2;
    for(int i=0; i<10; i++)
    {
        v1.push_back(i);
        v2.push_back(i+5);
    }

    vector<int> v_tar;
    //极端情况：大容器跟小容器没有一点交集，开辟空间时取两容器size之和即可
    v_tar.resize(v1.size() + v2.size());

    vector<int>::iterator pos_end = set_union(v1.begin(), v1.end(), v2.begin(), v2.end(), v_tar.begin());

    for_each(v_tar.begin(), pos_end, myprint);
    cout << endl;
}


void test563()
{
    /*
        V1={0~9}
        V2={5~14}

        V1和V2的差集={0 1 2 3 4}      V1中非V2的元素
        V2和V1的差集={10 11 12 13 14} V2中非V1的元素
    */

    vector<int> v1;
    vector<int> v2;
    for(int i=0; i<10; i++)
    {
        v1.push_back(i);
        v2.push_back(i+5);
    }

    vector<int> v_tar;
    //极端情况：大容器跟小容器没有一点交集，开辟空间时取两容器较大者size即可
    v_tar.resize(max(v1.size(), v2.size()));

    cout << "v1和v2的差集:" << endl;
    vector<int>::iterator pos_end = set_difference(v1.begin(), v1.end(), v2.begin(), v2.end(), v_tar.begin());
    for_each(v_tar.begin(), pos_end, myprint);
    cout << endl;

    cout << "v2和v1的差集:" << endl;
    pos_end = set_difference(v2.begin(), v2.end(), v1.begin(), v1.end(), v_tar.begin());
    for_each(v_tar.begin(), pos_end, myprint);
    cout << endl;
}


int main()
{
    test561();
    test562();
    test563();
    /*
        总结：

            求交集的两个集合必须的有序序列
            目标容器开辟空间需要从两个容器中取小值
            set_intersection返回值既是交集中最后一个元素的位置

            求并集的两个集合必须的有序序列
            目标容器开辟空间需要两个容器相加
            set_union返回值既是并集中最后一个元素的位置

            求差集的两个集合必须的有序序列
            目标容器开辟空间需要从两个容器取较大值
            set_difference返回值既是差集中最后一个元素的位置
    */

    system("pause");
    return 0;
}