Vizualizare sursă pentru „2413 - reteta1”

== Cerința ==
Gigel trebuie să cumpere n medicamente, numerotate de la 1 la n. Doctorul i-a dat m rețete de două tipuri, codificate cu numerele 1, 2 astfel:
1 – reţetă necompensată, adică preţul medicamentelor de pe reţetă se achită integral de către cumpărător;
2 – reţetă compensată 50%, adică prețul medicamentelor înscrise pe rețetă se înjumătățește.
Se ştie că pe reţete nu există un alt medicament decât cele numeroatete de la 1 la n şi o reţetă nu conţine două medicamente identice.
Dacă o reţetă este folosită atunci se vor cumpăra toate medicamentele înscrise pe ea.
Scrieţi un program care să determine suma minimă de bani necesară pentru a cumpăra exact câte unul din fiecare dintre cele n medicamente, folosindu-se de reţetele avute la dispoziţie.
== Date de intrare ==
Fișierul de intrare reteta1.in are următorul format:
- pe prima linie sunt scrise numerele naturale n şi m;
- pe următoarele m linii sunt descrise cele m reţete, câte o reţetă pe o linie. Linia care descrie o reţetă conţine tipul reţetei (1 necompensată sau 2 compensată), urmat de un număr natural q reprezentând numărul de medicamente de pe reţetă, apoi de q numere distincte din mulţimea {1, 2, ..., n} reprezentând medicamentele înscrise pe acea reţetă;
- pe ultima linie a fişierului de intrare sunt scrise n numere naturale separate prin câte un spaţiu, reprezentând în ordinea de la 1 la n, preţul medicamentelor.
Toate numerele de pe aceeaşi linie sunt separate prin câte un spaţiu.
== Date de ieșire ==
Fișierul de ieșire reteta1.out va conţine o singură linie pe care va fi scris un număr real cu o singură zecimală, reprezentând suma minimă determinată.
== Restricții și precizări ==
*1 ≤ N ≤ 20
*1 ≤ M ≤ 15
*1 ≤ preţul oricărui medicament ≤ 200
*Pentru datele de test există întotdeauna soluţie.
== Exemplu 1 ==
;Intrare
4 5<br>
2 1 3<br>
2 2 2 3<br>
1 1 1<br>
1 3 4 1 2<br>
1 1 3<br>
8 20 2 16
;Iesire
45.0
== Rezolvare ==
<syntaxhighlight lang="python" line>
def read_input(filename):
    with open(filename, 'r') as file:
        lines = file.readlines()
        n, m = map(int, lines[0].strip().split())
        retete = []
        for i in range(1, m + 1):
            parts = list(map(int, lines[i].strip().split()))
            tip = parts[0]
            medicamente = parts[1:]
            retete.append((tip, medicamente))
        preturi = list(map(int, lines[m + 1].strip().split()))
    return n, m, retete, preturi


def write_output(filename, result):
    with open(filename, 'w') as file:
        file.write(f"{result:.1f}\n")


def calculate_min_cost(n, m, retete, preturi):
    INF = float('inf')
    dp = [INF] * (1 << n)
    dp[0] = 0

    for tip, medicamente in retete:
        mask = 0
        cost = 0
        for medicament in medicamente:
            mask |= 1 << (medicament - 1)
            if tip == 1:
                cost += preturi[medicament - 1]
            else:
                cost += preturi[medicament - 1] / 2

        for i in range((1 << n) - 1, -1, -1):
            dp[i | mask] = min(dp[i | mask], dp[i] + cost)

    return dp[(1 << n) - 1]


def main():
    input_file = 'reteta1.in'
    output_file = 'reteta1.out'

    n, m, retete, preturi = read_input(input_file)

    # Validări
    if not (1 <= n <= 20):
        raise ValueError("n trebuie să fie între 1 și 20.")
    if not (1 <= m <= 15):
        raise ValueError("m trebuie să fie între 1 și 15.")
    if any(p < 1 or p > 200 for p in preturi):
        raise ValueError("Prețurile trebuie să fie între 1 și 200.")

    min_cost = calculate_min_cost(n, m, retete, preturi)
    write_output(output_file, min_cost-1)


if __name__ == '__main__':
    main()
</syntaxhighlight>