AF - Pb2 din Curs 10 - mulţimi

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

ifstream f("multimi.in");

void citire(vector<int>& vect, int lungime)
{

    for (int i = 0; i < lungime; i++)
    {
        int nr;
        f >> nr;
        vect.push_back(nr);
    }
}

vector<int> alegeSubmultime(vector<int>& a, vector<int>& c)
{
    vector<int> rezultat;

    int i = 0;

    while (i < c.size() && c[i] > 0)
        rezultat.push_back(a[i++]);

    i += a.size() - c.size();

    while (i < a.size())
        rezultat.push_back(a[i++]);

    return rezultat;
}

int main()
{
    int n, m;
    f >> n >> m;

    vector<int> c, a;

    citire(c, n);
    citire(a, m);

    sort(c.begin(), c.end(), greater<int>());
    sort(a.begin(), a.end(), greater<int>());

    vector<int> rez = alegeSubmultime(a, c);

    for (int x : rez)
        cout << x << ' ';
    return 0;
}