Hotel Room Allocation

Assign arriving groups in order. A group must stay entirely in the earliest hotel with enough free rooms. Subtract the group size from that hotel’s free rooms. If no hotel fits, return 0 for that group and leave all capacities unchanged. Hotel positions start at 1.

Implement hotel_room_allocation(rooms, groups).

Input

rooms: Initial free rooms per hotel, in priority order.

groups: Group sizes in arrival order.

Output

Return the selected hotel position for each group, or 0 when it cannot be placed.

Contraintes

1…200,000 hotels and 0…200,000 groups. Initial room counts are 0…1,000,000,000. Group sizes are 1…1,000,000,000. Empty groups input returns an empty list.

Exemples

Entrée :
rooms = [3, 1, 4]
groups = [2, 3, 2]

Sortie attendue :
expected = [1, 3, 0]

Entrée :
std::vector<long long> rooms = {3, 1, 4};
std::vector<long long> groups = {2, 3, 2};

Sortie attendue :
std::vector<long long> expected = {1, 3, 0};

Entrée :
long[] rooms = new long[] {3, 1, 4};
long[] groups = new long[] {2, 3, 2};

Sortie attendue :
long[] expected = new long[] {1, 3, 0};

Entrée :
rooms = [2, 2]
groups = [2, 2]

Sortie attendue :
expected = [1, 2]

Entrée :
std::vector<long long> rooms = {2, 2};
std::vector<long long> groups = {2, 2};

Sortie attendue :
std::vector<long long> expected = {1, 2};

Entrée :
long[] rooms = new long[] {2, 2};
long[] groups = new long[] {2, 2};

Sortie attendue :
long[] expected = new long[] {1, 2};

Références

Ce défi a été adapté pour CodeUtility. Suivez l'énoncé et les contraintes ci-dessus.

• CSES: Hotel Queries