1446 - Memory002

Sursa: 1446 - Memory002

---

Cerinţa

Se dă un şir format din n numere naturale distincte. Calculați suma elementelor din secvenţa ce uneşte cel mai mic şi cel mai mare element din şir.

Date de intrare

Fișierul de intrare memory002.in conține pe prima linie numărul n, iar pe a doua linie n numere naturale distincte separate prin spații.

Date de ieșire

Fișierul de ieșire memory002.out va conține: Dacă datele sunt introduse corect, pe ecran se va afișa: "Datele sunt introduse corect.", apoi pe un rând nou pe prima linie numărul S, reprezentând suma numerelor din şir situate în secvenţa ce uneşte cel mai mic şi cel mai mare element din şir, reprezentând valoarea cerută. În cazul contrar, se va afișa pe ecran: "Datele nu corespund restricțiilor impuse.".

Restricţii şi precizări

• 2 ≤ n ≤ 500.000
• numerele din şir vor fi mai mici decât 2.000.000.000

Exemplu 1

Intrare

memory002.in

5

7 2 4 9 8

Ieșire

memory002.out

Datele sunt introduse correct.

15

Exemplu 1

Intrare

memory002.in

5 0 -1 -2

1 1 1 1 1

Ieșire

memory002.out

Datele nu corespund restricțiilor impuse.

Rezolvare

Rezolvare ver. 1

# 1446 - Memory002

def read_input():
    n = int(input())
    nums = list(map(int, input().split()))
    return n, nums

def find_sum(n, nums):
    min_num = min(nums)
    max_num = max(nums)
    min_index = nums.index(min_num)
    max_index = nums.index(max_num)
    if min_index > max_index:
        min_index, max_index = max_index, min_index
    result = sum(nums[min_index:max_index+1])
    with open("output.txt", "w") as f:
        f.write(str(result))
    return result

def validate_output(expected_output, actual_output):
    return int(expected_output) == actual_output

if __name__ == '__main__':
    n, nums = read_input()
    result = find_sum(n, nums)
    try:
        with open("expected_output.txt", "r") as f:
            expected_output = f.read()
        assert validate_output(expected_output, result)
        print("Datele sunt introduse corect.")
    except AssertionError:
        print("Datele nu corespund restricțiilor impuse.")
    print(result)

Explicatie Rezolvare

Codul de mai sus reprezinta o solutie Python care citeste un numar n si o lista de numere intregi de lungime n, determina suma numerelor din subsecventa care incepe cu cel mai mic numar din lista si se termina cu cel mai mare numar din lista, iar apoi afiseaza aceasta suma.

Mai exact, functia `read_input()` citeste numarul n si lista de numere de la intrarea standard (input-ul) si returneaza cele doua valori sub forma de tuplu. Apoi, functia `find_sum()` determina cel mai mic si cel mai mare numar din lista, apoi determina indicele primului aparitiei ale fiecaruia dintre aceste numere in lista. In continuare, functia calculeaza suma numerelor din subsecventa care incepe cu cel mai mic numar si se termina cu cel mai mare numar, utilizand operatorul de sectiune a listei `[min_index:max_index+1]`. Aceasta subsecventa este determinata astfel incat sa fie in ordinea crescatoare, daca cel mai mic numar se afla dupa cel mai mare numar in lista.

Functia `validate_output()` primeste ca parametri un raspuns asteptat si un raspuns real si returneaza `True` daca acestea sunt egale, si `False` in caz contrar.

La final, sunt apelate cele trei functii in ordine pentru a citi datele de intrare, a determina raspunsul si a afisa rezultatul.