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.

Ràng buộc

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.

Ví dụ

Đầu vào:
rooms = [3, 1, 4]
groups = [2, 3, 2]

Kết quả mong đợi:
expected = [1, 3, 0]

Đầu vào:
std::vector<long long> rooms = {3, 1, 4};
std::vector<long long> groups = {2, 3, 2};

Kết quả mong đợi:
std::vector<long long> expected = {1, 3, 0};

Đầu vào:
long[] rooms = new long[] {3, 1, 4};
long[] groups = new long[] {2, 3, 2};

Kết quả mong đợi:
long[] expected = new long[] {1, 3, 0};

Đầu vào:
rooms = [2, 2]
groups = [2, 2]

Kết quả mong đợi:
expected = [1, 2]

Đầu vào:
std::vector<long long> rooms = {2, 2};
std::vector<long long> groups = {2, 2};

Kết quả mong đợi:
std::vector<long long> expected = {1, 2};

Đầu vào:
long[] rooms = new long[] {2, 2};
long[] groups = new long[] {2, 2};

Kết quả mong đợi:
long[] expected = new long[] {1, 2};

Tham khảo

Thử thách này đã được điều chỉnh cho CodeUtility. Hãy làm theo đề bài và các ràng buộc ở trên.

• CSES: Hotel Queries