1708 - Cuburi 3

Ionuţ a învăţat la şcoală să lucreze cu numere mari. El are la dispoziţie un şir de N numere naturale nenule. Din fiecare număr el şterge exact trei cifre, fără să schimbe ordinea cifrelor rămase, astfel încât să obţină cel mai mic număr natural nenul posibil. De exemplu, din numărul 20731049 se obţine numărul 20049, iar din numărul 13004 se obţine numărul 10. Înlocuind fiecare număr citit cu numărul obţinut prin operaţia de mai sus, Ionuţ obţine un nou şir şi scrie termenii acestuia pe feţele unor cuburi astfel: primele şase numere din şir le scrie pe primul cub şi îl notează pe acesta cu 1, următoarele şase numere din şir le scrie pe un alt cub pe care îl notează cu 2 ş.a.m.d.

Aceste cuburi au fost distribuite în piramide după modelul din figura. Piramidele au fost numerotate cu numere naturale consecutive. Piramida cu numărul de ordine 1 este formată numai din cubul cu numărul de ordine 1 şi are un singur nivel, piramida cu numărul de ordine 2 are pe primul nivel cuburile 2, 3 şi 4 iar pe ultimul nivel cubul 5 ş.a.m.d.

Două niveluri alăturate în cadrul unei piramide diferă prin exact două cuburi. Primul nivel al unei piramide conţine cu două cuburi mai mult decât primul nivel al piramidei precedente. Piramida se consideră completă dacă pe ultimul nivel are un singur cub.

Cerința

Scrieţi un program care citeşte numerele naturale nenule N şi K, apoi cele N numere naturale ce fac parte din şirul iniţial, şi determină:

a) Numărul de piramide complete construite de Ionuţ.

b) Numerele scrise pe cuburile din primele K piramide.

Date de intrare

Fişierul cuburi.in are două linii: prima linie conţine două numere naturale, N şi K, iar a doua linie conţine N numere naturale. Pe fiecare linie a fişierului numerele sunt separate prin câte un spaţiu.

Date de ieșire

Fişierul cuburi.out are două linii: prima linie conţine numărul de piramide complete care au fost construite, iar a doua linie conţine toate numerele scrise pe cuburile ce formează primele K piramide. Numerele sunt scrise separate prin câte un spaţiu, în ordinea apariţiei lor în şirul nou obţinut.

Restricții și precizări

• 6 ≤ N ≤ 100000
• Se garantează că se pot construi cel puţin K piramide complete.
• Cele N numere naturale de pe a doua linie a fişierului de intrare au minimum 4 cifre şi maximum 9 cifre.

Exemplu:

cuburi.in

31 2
18250 9280 18250 953805 20800 6040065 24208 4405 8794 1720 98886 96400 45544 8560056 36055 60400 80200 11560 36475 26992 68320 69400 20296 72640 34048 57700 66520 47440 91232 26080 90280

cuburi.out

2
10 2 10 305 20 4005 20 4 4 1 86 40 44 5005 30 40 20 10 34 22 20 40 20 20 30 50 20 40 12 20

Încărcare soluție

Lipește codul aici

import math

def nrmin(x, nrc):
    v = [int(digit) for digit in str(x)]
    Min = v[0]
    imin = 0
    for i in range(4):
        if v[i] < Min and v[i] != 0:
            Min = v[i]
            imin = i
    nr = Min
    in_index = imin + 1
    sf = 4
    while sf < nrc:
        Min = v[in_index]
        imin = in_index
        for i in range(in_index, sf + 1):
            if v[i] < Min:
                Min = v[i]
                imin = i
        nr = nr * 10 + Min
        in_index = imin + 1
        sf += 1
    return nr

def main():
    with open("cuburi.in", "r") as stdin, open("cuburi.out", "w") as stdout:
        n, k = map(int, stdin.readline().split())
        nr = sum(i * i * 6 for i in range(1, k + 1))
        w = []
        nrpir = 0
        s = 0
        for i in range(1, n + 1):
            x = int(stdin.readline())
            nrc = int(math.log10(x)) + 1
            if i % 6 == 0 and math.sqrt(i // 6) == int(math.sqrt(i // 6)):
                s += i
                if s <= n:
                    nrpir += 1
            if i <= nr:
                w.append(nrmin(x, nrc))
        stdout.write(f"{nrpir}\n")
        for num in w:
            stdout.write(f"{num} ")
        stdout.write("\n")

if __name__ == "__main__":
    main()