Vizualizare sursă pentru „4137 - rgb”

Ionuţ, tânăr programator, se lansează pe piaţa producătorilor de jocuri pe calculator. Jocul pe care l-a proiectat se numeşte RGB. În joc există <code>N</code> personaje extraterestre. Fiindcă Ionuţ nu este de acord cu teoria omuleţilor verzi, personajele lui sunt de trei culori:

* <code>R</code> extratereştri de culoare roşie;
* <code>G</code> extratereştri de culoare verde;
* <code>B</code> extratereştri de culoare albastră.

Fiecare extraterestru are o anumită putere, exprimată printr-un număr natural impar, puterile oricăror doi extratereştri fiind diferite. Pe parcursul jocului fiecare extraterestru va lupta cu fiecare dintre ceilalţi extratereştri. Rezultatul unei lupte între doi extratereştri depinde de puterea acestora, dar şi de culoarea lor. Într-o luptă dintre doi extratereştri de aceeaşi culoare, va câştiga cel cu puterea cea mai mare. Într-o luptă între doi extratereştri de culori diferite, puterile lor se modifică după cum urmează, iar după modificare lupta o va câştiga extraterestrul cu puterea mai mare:

* Puterea unui extraterestru roșu este dublată dacă adversarul este un extraterestru verde.
* Puterea unui extraterestru verde este dublată dacă adversarul este un extraterestru albastru.
* Puterea unui extraterestru albastru este dublată dacă adversarul este un extraterestru roșu.

După fiecare luptă, puterile extratereștrilor revin la valorile inițiale, în caz că s-au modificat.

= Cerința =
Scrieţi un program care, cunoscând culorile şi puterile extratereştrilor, rezolvă următoarele două cerinţe:

1) determină puterea extraterestrului care câştigă cele mai multe lupte; dacă există mai mulţi astfel de extratereştri, se va afişa puterea minimă;

2) determină pentru fiecare extraterestru numărul de lupte câştigate de acesta.

= Date de intrare =
Fișierul de intrare <code>rgb.in</code> conține pe prima linie numerele naturale <code>C</code> <code>R</code> <code>G</code> și <code>B</code>, unde <code>C</code> este cerinţa care trebuie să fie rezolvată (1 sau 2), <code>R</code> reprezintă numărul de extratereștri roșii, <code>G</code> numărul de extratereștri verzi, iar <code>B</code> numărul de extratereștri albaştri. Pe cea de a doua linie se află <code>R</code> numere naturale impare în ordine strict crescătoare, reprezentând puterile celor R extratereștri roșii. Pe cea de a treia linie se află <code>G</code> numere naturale impare în ordine strict crescătoare, reprezentând puterile celor G extratereștri verzi. 

Pe cea de a patra linie se află <code>B</code> numere naturale impare în ordine strict crescătoare, reprezentând puterile celor B extratereștri albaştri. Valorile scrise pe aceeaşi linie sunt separate prin câte un spaţiu.

= Date de ieșire =
Pentru <code>C = 1</code>, fişierul de ieşire <code>rgb.out</code> va conţine o singură linie pe care va fi scrisă puterea extraterestrului care câştigă cele mai multe lupte; dacă există mai mulţi extratereştri care câştigă un număr maxim de lupte, se va afişa puterea minimă.Pentru <code>C = 2</code>, fişierul de ieşire <code>rgb.out</code> va conţine trei linii. Pe prima linie se va scrie numărul de lupte câştigate de fiecare extraterestru roșu. Pe a doua linie, se va scrie numărul de lupte câştigate de fiecare extraterestru verde. Pe a treia linie, se va scrie numărul de lupte câştigate de fiecare extraterestru albastru. Pentru fiecare culoare, valorile vor fi afişate considerând ordinea extratereştrilor din fişierul de intrare. Valorile scrise pe aceeaşi linie vor fi separate prin câte un spaţiu.

= Restricții și precizări =

* <code>N = R + G + B</code>, <code>N ≤ 500.000</code>
* <code>1 ≤ R, G, B ≤ N-2</code>
* <code>1 ≤ puterea oricărui extraterestru ≤ 2 • N - 1</code>
* Datorită dimensiunilor mari, unele teste nu au fost adăugate

= Exemplul 1: =
<code>rgb.in</code>
 1 1 2 2
 3
 1 7
 5 9
<code>rgb.out</code>
 7

=== Explicație ===
<code>C = 1</code>, deci se va rezolva prima cerinţă. Există un extraterestru roşu, care are puterea <code>3</code>, doi extratereştri verzi, având puterile <code>1</code>, respectiv <code>7</code> şi doi extratereştri albaștri cu puterile <code>5</code>, respectiv <code>9</code>. Extraterestrul cu puterea <code>7</code> este singurul care va câştiga cele mai multe lupte (în cazul acesta, chiar toate):

* când luptă cu extraterestrul verde cu puterea <code>1</code> câştigă, pentru că are puterea mai mare;
* când luptă cu extraterestrul roşu cu puterea <code>3</code>, acesta îşi va dubla puterea (va avea puterea <code>6</code>), dar va fi insuficient pentru a câștiga lupta;
* când luptă contra extraterestrului albastru cu puterea <code>9</code>, va avea puterea dublată (<code>14</code>), prin urmare va câştiga şi această luptă.
<syntaxhighlight lang="python" line="1">
def determine_winner_and_counts(N, colors, powers):
    # Initialize the win counts
    win_counts = [0] * N
    
    for i in range(N):
        for j in range(i + 1, N):
            power_i = powers[i]
            power_j = powers[j]
            
            # Adjust powers based on the colors
            if colors[i] == 'R' and colors[j] == 'G':
                power_i *= 2
            elif colors[i] == 'G' and colors[j] == 'R':
                power_j *= 2
            elif colors[i] == 'G' and colors[j] == 'B':
                power_i *= 2
            elif colors[i] == 'B' and colors[j] == 'G':
                power_j *= 2
            elif colors[i] == 'B' and colors[j] == 'R':
                power_i *= 2
            elif colors[i] == 'R' and colors[j] == 'B':
                power_j *= 2
            
            # Determine winner
            if power_i > power_j:
                win_counts[i] += 1
            else:
                win_counts[j] += 1
    
    # Determine the extraterrestrial with the most wins
    max_wins = max(win_counts)
    winners = [powers[i] for i in range(N) if win_counts[i] == max_wins]
    min_power_winner = min(winners)
    
    return min_power_winner, win_counts

# Exemplu de utilizare
N = 5
colors = ['R', 'G', 'B', 'R', 'G']
powers = [5, 3, 7, 9, 1]

min_power_winner, win_counts = determine_winner_and_counts(N, colors, powers)
print("Puterea extraterestrului cu cele mai multe lupte câștigate:", min_power_winner)
print("Numărul de lupte câștigate de fiecare extraterestru:", win_counts)
</syntaxhighlight>