Editing 1061 - Cifru1 (section)

= Restricții și precizări =

* <code>1 < N ≤ 100 000</code>
* Un disc poate să rămână nemişcat.

== Exemplul 1 ==
input.txt:

4

7

3

9

0

output.txt:

9

7

0

2

Explicație:

Avem un cifru cu <code>4</code> discuri. Iniţial, cifrul este în starea <code>7390</code> (primul disc este poziţionat pe cifra <code>7</code>, al doilea pe cifra <code>3</code> etc.)

Cea mai mare cifră de pe cifru este cifra <code>9</code>.

Numărul minim de mutări este <code>7</code> şi se poate obţine în două moduri:

# Deplasăm primul disc cu <code>2</code> poziţii în sus, al doilea disc cu <code>4</code> poziţii în jos, al treilea rămâne nemişcat, iar ultimul se deplasează cu o poziţie în jos. Combinaţia obţinută este <code>9999</code>.
# Deplasăm primul disc cu <code>3</code> poziţii în sus, al doilea disc cu <code>3</code> poziţii în jos, al treilea cu o poziţie în sus, iar ultimul rămâne nemişcat. Combinaţia obţinută este <code>0000</code>.

Astfel, cifra cea mai mică ce formează combinaţia cu număr minim de mutări este <code>0</code>. Avem <code>2</code> combinaţii care se pot obţine în numărul minim de mutări determinat: <code>0000</code> şi <code>9999</code>.

== Exemplul 2 ==
input.txt:

99999999999999

7

3

9

0

Output:

Input-ul nu convine conditiilor

== Rezolvare ==
<syntaxhighlight lang="python3" line="1">
def verificare(n):
    if not(1<=n<=100000):
        print("Input-ul nu convine conditiilor")
        exit()

Fin = "input.txt"
Fou = "output.txt"

with open(Fin, "r") as IN, open(Fou, "w") as OUT:
    N = int(IN.readline())
    verificare(N)
    Apar = [0] * 10  # Apar[i] = 1 if color i appears on at least one disc
    MAX = 0          # maximum digit
    NrMin = 10 * N + 1  # minimum number of moves
    Cif = -1         # digit obtained in the minimum number of moves
    Cate = 0         # number of possibilities

    # initialize
    for i in range(10):
        Apar[i] = 0

    # read input data and determine initially appearing digits and maximum digit
    for i in range(1, N + 1):
        x = int(IN.readline())
        Apar[x] += 1
        if MAX < x:
            MAX = x

    # calculate the number of moves for each appearing digit
    for i in range(10):
        Nr = 0
        for j in range(10):
            if Apar[j] and j != i:
                Nr += min(abs(j - i), 10 - abs(j - i)) * Apar[j]

        if Nr < NrMin:
            NrMin = Nr
            Cif = i
            Cate = 1
        elif Nr == NrMin:
            Cate += 1

    OUT.write(f"{MAX}\n{NrMin}\n{Cif}\n{Cate}\n")

</syntaxhighlight>