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.

制約

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.

例

入力：
rooms = [3, 1, 4]
groups = [2, 3, 2]

期待される出力：
expected = [1, 3, 0]

入力：
std::vector<long long> rooms = {3, 1, 4};
std::vector<long long> groups = {2, 3, 2};

期待される出力：
std::vector<long long> expected = {1, 3, 0};

入力：
long[] rooms = new long[] {3, 1, 4};
long[] groups = new long[] {2, 3, 2};

期待される出力：
long[] expected = new long[] {1, 3, 0};

入力：
rooms = [2, 2]
groups = [2, 2]

期待される出力：
expected = [1, 2]

入力：
std::vector<long long> rooms = {2, 2};
std::vector<long long> groups = {2, 2};

期待される出力：
std::vector<long long> expected = {1, 2};

入力：
long[] rooms = new long[] {2, 2};
long[] groups = new long[] {2, 2};

期待される出力：
long[] expected = new long[] {1, 2};

参考資料

この課題はCodeUtility用に調整されています。上記の説明と制約に従ってください。

• CSES: Hotel Queries