3698 - Bemo

Se dă o matrice cu R linii şi C coloane de numere distincte de la 1 la R * C. Bemo, personajul emoţional, doreşte să urmărească cel mai bun drum din colţul superior stânga, de coordonate (1, 1), în colţul inferior dreapta, de coordonate (R, C). Un drum este o secvenţă de numere din matrice în care fiecare număr se găseşte în jos sau la dreapta numărului anterior, adică dacă (i, j) este poziţia unui număr de pe un drum, atunci următorul număr poate fi cel de pe poziţia (i + 1, j) sau cel de pe poziţia (i, j + 1). Pentru a determina dacă un drum A este mai bun decât un drum B, numerele fiecărui drum se vor sorta şi se va alege cel mai mic lexicografic, de exemplu [1,3,5,6,8] < [1,4,5,6,7].

Date de intrare

Fișierul de intrare bemoin.txt conține pe prima linie două numere naturale R și C, unde R este numărul liniilor, iar C numărul coloanelor matricei lui Bemo. Pe următoarele R linii se vor găsi câte C numere separate printr-un spaţiu. Fiecare număr va fi unic şi va fi cuprins în intervalul [1, R*C].

Date de ieșire

Fișierul de ieșire bemoout.txt va conține R+C-1 numere reprezentând cel mai bun drum pe care Bemo îl poate alege. Numerele vor fi scrise separate printr-un spațiu.

Restricții și precizări

• 0 < R, C < 1501;
• Pentru 40% din teste: 0 < R, C < 751;
• Pentru 70% din teste: 0 < R, C < 1301;
• Spunem că un drum A = (a1, a2, …, aR+C-1) este mai mic lexicografic decât un drum B = (b1, b2,…, bR+C-1) dacă există o poziție p astfel încât ap < bp și a1 = b1, a2 = b2, …, ap-1 = bp-1.

Exemplul 1

bemoin.txt

4 4

7 4 13 3

8 11 12 2

10 9 1 5

16 14 15 6

bemoout.txt

7 4 11 9 1 5 6

Exemplul 2

bemoin.txt

3 3

5 1 4

8 7 6

9 2 3

bemoout.txt

5 1 4 7 6 3

Rezolvare

#3698 - Bemo
def find_best_path(R, C, matrix):
    if not (0 < R < 1501 and 0 < C < 1501):
        return "Fals"

    path = []

    i, j = 0, 0

    while i < R and j < C:
        path.append(matrix[i][j])

        if i < R - 1 and (j == C - 1 or matrix[i + 1][j] < matrix[i][j + 1]):
            i += 1
        else:
            j += 1

    return path

# Citim datele de intrare din fișierul "bemoin.txt"
with open("bemoin.txt", "r") as file:
    R, C = map(int, file.readline().split())
    matrix = [list(map(int, file.readline().split())) for _ in range(R)]

# Găsim cel mai bun drum
result = find_best_path(R, C, matrix)

# Scriem rezultatul în fișierul "bemoout.txt" sau afișăm "Fals" în cazul în care restricțiile nu sunt respectate
with open("bemoout.txt", "w") as file:
    if isinstance(result, list):
        file.write(" ".join(map(str, result)))
    else:
        file.write(result)