View source for 0554 - Marte

==Cerința==
Pe planeta Marte banii sunt altfel decât la noi. Există n tipuri de monede și pentru fiecare se cunoaște valoarea v[i], exprimată în lei marțieni. De asemenea, există m marțieni, numerotați de la 1 la m și fiecare are un anumit număr de monede din fiecare tip. Să se determine care dintre marțieni este cel mai bogat, adică acel marțian care are număr total de lei marțieni maxim. Dacă există mai mulți cei mai bogați marțieni, se va determina marțianul cu numărul de ordine cel mai mic.
==Date de intrare==
Programul citește de la tastatură numărul n, iar apoi n numere naturale, reprezentând valoarea fiecărei monede marțiene. Urmează numărul de marțieni m, apoi m seturi cu câte n numere, reprezentând numărul de monede din fiecare tip pe care îl are, în ordine, fiecare marțian.
==Date de ieșire==
Programul va afișa pe ecran numărul X, reprezentând numărul de ordine al celui mai bogat marțian.
==Restricții și precizări==
*1 ≤ n ≤ 50;
*1 ≤ v[i] ≤ 5000;
*1 ≤ m ≤ 1000;
*fiecare marțian are cel mult 200 monede din fiecare tip – controlul averii este foarte strict pe planeta Marte;
==Exemplul 1==
;Intrare
:2
:2 5
:5
:3 2
:6 1
:5 2
:0 2
:1 3
;Ieșire
:Datele de intrare corespund restricțiilor impuse.
:3
==Exemplul 2==
;Intrare
:0 10
:4 7 3 6 7 1
;Ieșire
:Datele de intrare nu corespund restricțiilor impuse.
==Rezolvare==
<syntaxhighlight lang="python" line="">
#0554 - Marte
# defineste o functie ce calculeaza suma valorii monedelor, pentru fiecare martian
def calculeaza_suma(valori_monede, numar_monede):
    suma = 0
    for i in range(len(valori_monede)):
        suma += valori_monede[i] * numar_monede[i]
    return suma

# defineste o functie ce gaseste cel mai bogat martian, bazat pe suma valorii mondelor pe care le au 
def cel_mai_bogat_martian(n, valori_monede, m, info_martieni):
    cel_mai_bogat = 1
    # initializeaza maximul sumei pentru valorile monedelor
    suma_maxima = calculeaza_suma(valori_monede, info_martieni[0])
    for i in range(1, m):
        # calculeaza suma valorii monedelor pentru martianul curent
        suma_curenta = calculeaza_suma(valori_monede, info_martieni[i])
        # daca suma valorii monedelor pentru martianul curent e mai mare decat maximul sumei,
        # face update cu cel mai bogat martian si maximul sumei
        if suma_curenta > suma_maxima:
            cel_mai_bogat = i + 1
            suma_maxima = suma_curenta
    return cel_mai_bogat

if __name__ == '__main__':
    # citeste tipurile de monede, valorile lor, numarul martienilor si informatiile despre ei
    n = int(input("Introduceți numărul de tipuri de monede: "))
    valori_monede = list(map(int, input("Introduceți valorile monedelor: ").split()))
    m = int(input("Introduceți numărul de marțieni: "))
    info_martieni = []
    for i in range(m):
        info_martian = list(map(int, input(f"Introduceți numărul de monede pentru marțieni: ").split()))
        info_martieni.append(info_martian)
    # verifica daca se respecta restrictiile
    if n < 1 or n > 50 or m < 1 or m > 1000 or max(valori_monede) > 5000:
        print("Datele de intrare nu corespund restricțiilor impuse.")
    else:
        print("Datele de intrare corespund restricțiilor impuse.")
        # gaseste cel mai bogat martian
        X = cel_mai_bogat_martian(n, valori_monede, m, info_martieni)
        # afiseaza numarul celui mai bogat martian
        print(f"Cel mai bogat marțian este {X}.")
    
</syntaxhighlight>