4130 - Rotire25

Sursa: [1]

Cerinţa

Dându-se numerele X și K determinaţi:

1) Produsul dintre ultima cifră a numărului X * X * X * ... * X (de K ori) şi prima cifră a lui X.

2) Numărul rezultat după aplicarea celor K transformări.

Date de intrare

Pe prima linie a fișierului de intrare rotire25.in se află trei numere separate prin câte un spațiu C, X și K. Dacă C=1 se va rezolva doar prima cerință, iar dacă C=2 se va rezolva doar a doua cerință.

Date de ieșire

Fișierul sss.out conține un singur număr care reprezintă valoarea calculată conform cerinței, iar în consolă se va afișa mesajul "Datele introduse corespund cerintelor.". În caz contrar în consolă se va afișa mesajul "Datele introduse nu corespund cerintelor."

Restricţii şi precizări

• 1 ⩽ X ⩽ 999
• 1 ⩽ K ⩽ 1.000.000.000
• Pentru teste în valoare de 29 de puncte, C = 1.
• Pentru teste în valoare de 71 de puncte, C = 2.

Exemplul 1

Intrare

sss.in

1 6

120

4

21

5

31

6

Ieșire

Datele introduse corespund cerintelor.

sss.out

37

Exemplul 2

Intrare

sss.in

0 0

120

4

21

5

31

6

Ieșire

Datele introduse nu corespund cerintelor.

Rezolvare

<syntaxhighlight lang="python" line> def validate_input(C, N, a, K):

   if not (1 <= N <= 100000):
       print("Datele introduse nu corespund cerintelor")
       exit()
   if not all(0 < x <= 100000 for x in a[1:]):
       print("Datele introduse nu corespund cerintelor")
       exit()
   if C == 1 and len(str(a[1])) > 1 and not any(x.isdigit() for x in str(a[1])):
       print("Datele introduse nu corespund cerintelor")
       exit()

   print("Datele introduse corespund cerintelor.")

def read_input_file(filename):

   with open(filename, "r") as f:
       C, N = map(int, f.readline().split())
       a = [0] * (N+1)
       for i in range(1, N+1):
           a[i] = int(f.readline())
   return C, N, a

def write_output_file(filename, value):

   with open(filename, "w") as f:
       f.write(str(value))

def get_sum_last_n_nonzero_digits(C, N, a):

   def vcmddcn(n):
       while n:
           if n % 10:
               return n % 10
           else:
               n //= 10
       return -1
   c = vcmddcn(a[1])
   S = sum(a[N-c+1:])
   return S if C == 1 and c > 0 else None

def get_max_sum_subsequence(C, N, a):

   if C == 1:
       return None
   dp = [0] * (N+1)
   for i in range(1, N+1):
       S = 0
       for j in range(i-1, -1, -1):
           S += a[j+1]
           if i-j != S:
               dp[i] = max(dp[i], dp[j] + S)
           else:
               break
   return max(dp)

if __name__ == '__main__':

   C, N, a = read_input_file("sss.in")
   validate_input(C, N, a, K=2)
   result = get_sum_last_n_nonzero_digits(C, N, a) or get_max_sum_subsequence(C, N, a)
   write_output_file("sss.out", result)

</syntaxhighlight>

Explicatie rezolvare

Programul primește datele de intrare dintr-un fișier de intrare și le validează prin intermediul funcției validate_input. Dacă datele de intrare sunt corecte, atunci funcția get_sum_last_n_nonzero_digits sau funcția get_max_sum_subsequence este apelată în funcție de valoarea variabilei C, iar rezultatul este scris într-un fișier de ieșire.

Mai exact, dacă C este egal cu 1, atunci se va apela funcția get_sum_last_n_nonzero_digits pentru a calcula suma ultimelor n cifre nenule din primul element al listei a sau funcția get_max_sum_subsequence pentru a calcula suma maximă a unei subsecvențe nevide din lista a. Dacă C este diferit de 1, atunci se va apela doar funcția get_max_sum_subsequence pentru a calcula suma maximă a unei subsecvențe nevide din lista a.

Funcția validate_input se asigură că datele de intrare corespund cerințelor problemei, iar dacă nu corespund, atunci programul se va opri și va fi afișat un mesaj corespunzător. Funcția read_input_file citește datele de intrare din fișierul specificat, iar funcția write_output_file scrie rezultatul în fișierul specificat.