Editing 1204 - Trenuri (section)

== Rezolvare == 
<syntaxhighlight lang="python" line>
#1204 - Trenuri
def trenuri(N, M, trenuri, pasageri):
    # Verificare restricții
    if not (1 <= N <= 100000 and 1 <= M <= 100000):
        return "Fals"

    for i in range(N):
        if not (1 <= trenuri[i][0] <= trenuri[i][1] <= 1000000000):
            return "Fals"

    for i in range(M):
        if not (1 <= pasageri[i][0] <= pasageri[i][1] <= 1000000000):
            return "Fals"

    # Sortăm pasagerii în ordinea crescătoare a stației de sosire
    pasageri = sorted(pasageri, key=lambda x: x[1])

    # Sortăm trenurile în ordinea crescătoare a stației de sosire
    trenuri = sorted(enumerate(trenuri), key=lambda x: x[1][0])

    result = [0] * M  # Configurația trenurilor pentru fiecare pasager
    total_pasageri = 0  # Numărul total de pasageri

    for pasager in pasageri:
        start, end = pasager

        # Găsim primul tren disponibil în intervalul dorit
        for i, (statie, capacitate) in trenuri:
            if statie <= start and start <= end <= capacitate:
                result[pasager[2]] = i + 1  # Salvăm indexul trenului pentru pasager
                total_pasageri += 1
                break

    return total_pasageri, result

# Citirea datelor de intrare din fișierul "trenuriin.txt"
try:
    with open("trenuriin.txt", "r") as file:
        N, M = map(int, file.readline().split())
        trenuri = [tuple(map(int, file.readline().split())) for _ in range(N)]
        pasageri = [tuple(map(int, file.readline().split())) + (i,) for i in range(M)]
except Exception as e:
    print("Fals")
    exit()

# Calcularea numărului maxim de pasageri și a configurației trenurilor
result = trenuri(N, M, trenuri, pasageri)

if result == "Fals":
    print(result)
else:
    # Scrierea rezultatului în fișierul "trenuriout.txt"
    with open("trenuriout.txt", "w") as file:
        file.write(str(result[0]) + "\n")
        for tren in result[1]:
            file.write(str(tren) + "\n")

</syntaxhighlight>