Vizualizare sursă pentru „2976 - maxim7”

Dintr-un șir format din <code>N</code> cifre, numerotate de la <code>1</code> la <code>N</code>, Ionel ia exact <code>M</code> cifre aflate pe poziții consecutive. El lipește cifrele luate sau le amestecă și apoi le lipește pentru a obține cu ele un număr cât mai mare.

= Cerința =
Cunoscând <code>N</code>, <code>M</code> și cele <code>N</code> cifre din șir, să se determine:

1) cel mai mare număr care se poate obține din primele <code>M</code> dintre cele <code>N</code> cifre date;

2) de unde va lua Ionel <code>M</code> cifre aflate pe poziții consecutive pentru a obține un număr maxim; dacă sunt mai multe poziții corespunzătoare unui număr maxim, alegerea se va face astfel încât numărul format din cifrele rămase, în ordinea în care erau, să fie cât mai mare posibil; dacă și în acest caz există mai multe soluții, se alege poziția maximă.

= Date de intrare =
Din fișierul <code>maxim.in</code> se citesc: <code>P</code> de pe prima linie, reprezentând cerința problemei (<code>1</code> sau <code>2</code>), <code>N</code> și <code>M</code> de pe a doua linie, despărțite printr-un spațiu, cu semnificația din enunț, iar de pe linia a treia, se citesc cele <code>N</code> cifre, despărțite prin câte un spațiu.

= Date de ieșire =
În fișierul <code>maxim.out</code> se scrie:

- pentru <code>P = 1</code>: numărul maxim care se poate obține cu ajutorul primelor <code>M</code> cifre dintre cele <code>N</code> date, fără spații între cifrele numărului;

- pentru <code>P = 2</code>: un număr reprezentând poziția cerută.

= Restricții și precizări =

* <code>M</code>, <code>N</code> numere naturale, <code>1 ≤ N ≤ 500000</code>, <code>1 ≤ M ≤ 1000</code>, <code>M < N</code>
* Cele <code>N</code> valori de pe linia a treia sunt numere naturale între <code>0</code> și <code>9</code>
* Secvența de <code>N</code> cifre poate să înceapă cu cel mult <code>M - 1</code> cifre nule.
* <code>30</code> de puncte se vor obține cu rezolvarea cerinței <code>1</code>, iar <code>60</code> de puncte se vor obține cu rezolvarea cerinței <code>2</code>.
* Pentru <code>50%</code> dintre teste, <code>N < 1000</code> și <code>M < 10</code>.
* În concurs s-au acordat <code>10</code> puncte din oficiu. Aici se acordă pentru exemplele din enunț.

= Exemplul 1: =
<code>maxim.in</code>
 1
 10 3
 7 2 8 1 0 0 4 7 8 1
<code>maxim.out</code>
 872

=== Explicație ===
Se rezolvă cerința <code>1</code>. Cu primele trei cifre dintre cele <code>10</code> cifre date se pot forma numerele: <code>728</code>, <code>782</code>, <code>287</code>, <code>278</code>, <code>827</code>, <code>872</code>, cel mai mare fiind <code>872</code>.

= Exemplul 2: =
<code>maxim.in</code>
 2

 10 3
 7 2 8 1 0 0 4 7 8 1
<code>maxim.out</code>
 7

=== Explicație ===
Se rezolvă cerința <code>2</code>. Alegând cifrele începând de la poziția a <code>7</code>-a (cifrele <code>4</code>, <code>7</code> și <code>8</code>), se va forma numărul <code>874</code>, care este cel mai mare posibil.

== Încărcare soluție ==

=== Lipește codul aici ===
<syntaxhighlight lang="python" line="1">
import array as arr

dim = 500008
v = arr.array('h', [0] * dim)

w = [0] * 10
W = [0] * 10


def main():
with open("maxim.in", "r") as f, open("maxim.out", "w") as g:
    N, i, j, poz, M, x, z, k, ii = 0, 0, 0, 0, 0, 0, 0, 0, 0
    ok = 1
    p = int(f.readline())

    input_data = f.readline().split()
    N = int(input_data[0])
    M = int(input_data[1])

    v[1:N+1] = map(int, f.readline().split())

    for i in range(1, N+1):
        x = v[i]
        w[x] += 1

    if p == 1:
        if w[0] == M:
            g.write("0")
        else:
            for j in range(9, -1, -1):
                for i in range(1, w[j]+1):
                    g.write(str(j))

    if p == 2:
        for j in range(10):
            W[j] = w[j]

        for i in range(M+1, N+1):
            x = v[i]
            w[x] += 1
            z = v[i-M]
            w[z] -= 1
            ok = 1
            for j in range(9, 0, -1):
                if W[j] != w[j]:
                    if w[j] > W[j]:
                        ok = 2
                        W = w[:]
                        poz = i - M + 1
                    else:
                        ok = 0
                    break
            if ok == 1:
                poz = i - M + 1

        g.write(str(poz))

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