Chương trình tìm phòng phù hợp đầu tiên từ danh sách phòng trong Python

Cho danh sách rooms (dung tích các phòng) và giá trị mục tiêu t. Tìm phòng đầu tiên có dung tích ≥ t. Nếu không có, trả về -1.

Ví dụ: rooms = [20, 15, 35, 55, 30], t = 30 → kết quả 35 (phòng đầu tiên đủ lớn).

Thuật toán giải quyết (O(n))

1. Duyệt lần lượt từng room trong rooms
2. Nếu room >= t: trả về room
3. Kết thúc vòng lặp: trả về -1

Ví dụ minh họa

def solve(rooms, t):
    for room in rooms:
        if room >= t:
            return room
    return -1

rooms = [20, 15, 35, 55, 30]
t = 30
print(solve(rooms, t))

Đầu vào

[20, 15, 35, 55, 30], 30

Đầu ra

35