Написать количество предприятий типа А и Б через К лет.

#include <iostream>
#include <cstdlib>
#include <ctime>
#include <vector>
#include <algorithm>
#include <functional>
using namespace std;
 
inline int getAPeriod() { return rand() % 5 + 1; }
inline int getBPeriod() { return rand() % 3 + 2; }
 
int main()
{
//  srand( time( 0 ) );
 
    int H, M;
    int K;
    vector< int > A;
    vector< int > B;
 
    cin >> H >> M >> K;
 
    for ( int j = 0; j < H; ++j )
        A.push_back( getAPeriod() );
    make_heap( A.begin(), A.end(), greater< int >() );
    for ( int j = 0; j < M; ++j )
        B.push_back( getBPeriod() );
    make_heap( B.begin(), B.end(), greater< int >() );
 
    for ( int i = 1; i < K; ++i )
    {
        while ( A.size() > 0 && A.front() == i )
        {
            pop_heap( A.begin(), A.end(), greater< int >() );
            A.pop_back();
            B.push_back( getBPeriod() + i );
            push_heap( B.begin(), B.end(), greater< int >() );
        }
        while ( B.size() > 0 && B.front() == i )
        {
            pop_heap( B.begin(), B.end(), greater< int >() );
            B.back() = getBPeriod() + i;
            push_heap( B.begin(), B.end(), greater< int >() );
            A.push_back( getAPeriod() + i );
            push_heap( A.begin(), A.end(), greater< int >() );
        }
    }
 
    cout << "A: " << A.size() << endl;
    cout << "B: " << B.size() << endl;
 
    return 0;
}

#include <iostream>
#include <cstdlib>
using namespace std;
 
inline int getAPeriod() { return rand() % 5 + 1; }
inline int getBPeriod() { return rand() % 3 + 2; }
 
int main()
{
    int H, M, K;
 
    cin >> H >> M >> K;
 
    int *AtoB = new int [ K + 5 ]; //магическое число 5 зависит от периодов преобразования предприятий (максимальный срок)
    int *BtoAB = new int [ K + 5 ];
    int A = H;
    int B = M;
 
    for ( int i = 0; i < K; ++i ) { AtoB[ i ] = 0, BtoAB[ i ] = 0; }
    for ( int j = 0; j < H; ++j ) ++AtoB[ getAPeriod() ];
    for ( int j = 0; j < M; ++j ) ++BtoAB[ getBPeriod() ];
 
    for ( int i = 0, m; i < K; ++i )
    {
        m = rand() % 4;
        for ( int j = 0; j < m; ++j ) // добавляем 0-3 предприятия типа А
            ++AtoB[ getAPeriod() + i ];
        A += m;
        for ( int j = 0; j < AtoB[ i ]; ++j ) // преобразуем предприятия типа А в В
        {
            ++BtoAB[ getBPeriod() + i ];
        }
        A -= AtoB[ i ];
        B += AtoB[ i ];
        for ( int j = 0; j < BtoAB[ i ]; ++j ) // делим предприятие типа В на А и В
        {
            ++BtoAB[ getBPeriod() + i ];
            ++AtoB[ getAPeriod() + i ];
        }
        A += BtoAB[ i ];
    }
 
    cout << "A: " << A
        << "\nB: " << B << endl;
 
    return 0;
}