试题 算法训练 P0705

资源限制

内存限制：256.0MB C/C++时间限制：1.0s Java时间限制：3.0s Python时间限制：5.0s

　　输入两个整数集合A、B，求出他们的交集、并集以及B在A中的余集。交集、并集和余集的计算都要求写成一个单独的函数。

　　输入第一行为一个整数n，表示集合A中的元素个数。

　　第二行有n个按从小到大的顺序输入且互不相同的整数，表示集合A中的元素

　　第三行为一个整数m，表示集合B中的元素个数。

　　第四行有m个按从小到大的顺序输入且互不相同的整数，表示集合B中的元素

　　集合中的所有元素均为int范围内的整数，n、m<=1000。

　　输出第一行按从小到大的顺序输出A、B交集中的所有元素。

　　第二行按从小到大的顺序输出A、B并集中的所有元素。

　　第三行按从小到大的顺序输出B在A中的余集中的所有元素。

输入：

　　5

　　1 2 3 4 5

　　5

　　2 4 6 8 10

　　输出:

　　2 4

　　1 2 3 4 5 6 8 10

　　1 3 5

#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
#include <cmath>
#include <algorithm>
using namespace std;


int A[1005], B[1005], C[1005];


int main()
{
    int m, n;
    cin>>m;
    for(int i=0; i<m; i++)
        cin>>A[i];
    cin>>n;
    for(int i=0; i<n; i++)
        cin>>B[i];

    int ia=0, ib=0;
    while(!(ia==m && ib==n))
    {
        if(A[ia] == B[ib])
        {
            cout<<A[ia]<<" ";
            C[ia] = 1;

        }

        if(A[ia] >= B[ib] && ib < n)
            ib++;
        else if(A[ia] <= B[ib] && ia < m)
            ia++;
        else if(ia == m)
            ib++;
        else if(ib == n)
            ia++;
    }
    cout<<endl;

    ia=0, ib=0;
    while(!(ia==m && ib==n))
    {
        if(A[ia] > B[ib] && ib < n)
        {
            cout<<B[ib]<<" ";
            ib++;
        }
        else if(A[ia] < B[ib] && ia < m)
        {
            cout<<A[ia]<<" ";
            ia++;
        }
        else if(A[ia] == B[ib])
        {
            cout<<A[ia]<<" ";
            if(ia < m) ia++;
            if(ib < n) ib++;
        }
        else if(ia == m)
        {
            cout<<B[ib]<<" ";
            ib++;
        }
        else if(ib == n)
        {
            cout<<A[ia]<<" ";
            ia++;
        }

    }
    cout<<endl;

    for(int i=0; i<m; i++)
    {
        if(!C[i])
            cout<<A[i]<<" ";
    }

    return 0;
}