2266 - Calc Mare

Enunt

Un celebru (sperăm) rezolvitor de pe acest site își va lua un calculator nou dacă intră la liceul la care dorește să intre. Dar momentan rămâne pe calculatorul său rezolvând probleme. Una dintre ele era foarte simplă și constă în adunarea/scăderea a două numere naturale.

Cerința

Se dau 2 numere naturale. Calculați suma / diferenţa lor.

Date de intrare

Fișierul de intrare calcmarein.txt conține pe prima linie numărul a, pe a doua linie numărul b, iar pe a treia linie un semn + sau - care arată dacă numerele trebuie adunate / scăzute.

Date de ieșire

Fișierul de ieșire calcmareout.txt va conține pe prima linie numărul R, reprezentând rezultatul determinat.

Restricții și precizări

• 1 ⩽ nr. cifre număr ⩽ 2000.
• Pentru 50% din punctaj semnul va fi + , iar pentru restul de 50%, semnul -.
• Pentru 20% din punctaj nr. cifre număr ≤ 15.
• Se garantează că a≥b pentru scăderi.

Exemplu 1

calcmarein.txt

123

321

+

calcmareout.txt

444

Explicație

În fișierul de intrare sunt 2 numere care trebuie adunate, iar suma lor este 444.

Exemplu 2

calcmarein.out

-2

-1

%

calcmareout.txt

Nu au fost respectate cerintele impuse

Rezolvare

#2266 - CalcMare
def check_restrictions(a, b, operation):
    if not (1 <= len(str(a)) <= 2000 and 1 <= len(str(b)) <= 2000):
        return False

    if operation not in ('+', '-'):
        return False

    if len(str(a)) <= 15 and len(str(b)) <= 15:
        return True

    return True if operation == '+' else (a >= b)

def perform_operation(a, b, operation):
    if operation == '+':
        return a + b
    elif operation == '-':
        return a - b

# Citire date de intrare
try:
    with open("calcmarein.txt", "r") as file:
        a = int(file.readline().strip())
        b = int(file.readline().strip())
        operation = file.readline().strip()
except FileNotFoundError:
    print("Fisierul de intrare nu exista!")
    exit()

# Verificare daca sunt respectate cerintele
if not check_restrictions(a, b, operation):
    print("Nu au fost respectate cerintele impuse!")
    exit()

# Efectuare operație
result = perform_operation(a, b, operation)

# Scriere rezultat în fișierul de ieșire
with open("calcmareout.txt", "w") as file:
    file.write(str(result) + "\n")