Editing 2897 - numere26

Se consideră un șir format din <code>n</code> numere naturale. Asupra numerelor din șir se face următoarea prelucrare: fiecare valoare este înlocuită cu cel mai mare divizor prim al său. În noul șir se formează secvențe de numere care încep și se termină cu aceeași valoare, numite secvențe neuniforme.

= Cerința =
Cunoscând numerele naturale <code>n</code> și <code>c</code>, și un șir de <code>n</code> numere naturale, se cere să se rezolve următoarele cerințe:

1. dacă <code>c = 1</code>, atunci se cere să se afișeze lungimea maximă a unei secvențe neuniforme.

2. dacă <code>c = 2</code>, atunci se cere să se afișeze numărul total de secvențe neuniforme din șir.

= Date de intrare =
Fișierul de intrare <code>input.txt</code> conține pe prima linie, despărțite prin câte un spațiu, numerele naturale <code>n</code> și <code>c</code>, cu semnificația din enunț. A doua linie conține <code>n</code> numere naturale, despărțite prin câte un spațiu.

= Date de ieșire =
Dacă <code>c = 1</code>, atunci pe prima linie a fișierului <code>output.txt</code> va fi scris un singur număr ce reprezintă lungimea maximă a unei secvențe neuniforme. Dacă <code>c = 2</code>, atunci fișierul <code>output.txt</code> va conține un singur număr ce reprezintă numărul total de secvențe neuniforme.

= Restricții și precizări =

* <code>0 < n < 10.000</code>

== Exemplul 1 ==
input.txt:

6  1

14 2 49 3 35 1024  

output.txt:

5

Explicație:

Cele <code>10</code> numere sunt înlocuite cu valorile: <code>7</code>, <code>2</code>, <code>3</code>, <code>5</code>, <code>3</code>, <code>3</code>, <code>5</code>, <code>2</code>, <code>5</code>, <code>2</code>. Numărul total de secvențe neuniforme din șir este <code>3 + 3 + 3 = 9</code>.

== Exemplul 2 ==
input.txt:

10 2

14 8 3 25 6 24 20 1024 100 2

output.txt:

9

Explicație:

Cele <code>10</code> numere sunt înlocuite cu valorile: <code>7</code>, <code>2</code>, <code>3</code>, <code>5</code>, <code>3</code>, <code>3</code>, <code>5</code>, <code>2</code>, <code>5</code>, <code>2</code>. Numărul total de secvențe neuniforme din șir este <code>3 + 3 + 3 = 9</code>.

== Exemplul 3 ==
input.txt:

999999999999 2

14 8 3 25 6 24 20 1024 100 2

Output:

Input-ul nu convine conditiilor

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

def fact_max(a):
    f = 2
    r = 0
    if a % 2 == 0:
        r = 2
        while a % 2 == 0:
            a //= 2
    for f in range(3, a + 1, 2):
        if a % f == 0:
            r = f
            while a % f == 0:
                a //= f
    return r

def main():
    with open("input.txt", "r") as fin, open("output.txt", "w") as fout:
        s = 0
        n, c = map(int, fin.readline().split())
        verificare(n)
        fr1 = [0] * 10000
        fr2 = [0] * 10000
        mx = 0
        lmax = 0

        l=list(map(int, fin.readline().split()))

        for i in range(1, n + 1):
            a = l[i-1]
            x = fact_max(a)

            if x > mx:
                mx = x  # cel mai mare nr prim care apare in urma prelucrarii sirului initial

            # cerinta 1
            if fr1[x] == 0:
                fr1[x] = i
            else:
                d = i - fr1[x] + 1
                if d > lmax:
                    lmax = d

            # cerinta 2
            fr2[x] += 1

        if c == 1:
            fout.write(str(lmax) + "\n")
        else:
            # rezolvarea de la cerinta 2
            # calculez suma de x*(x-1)/2, unde x=fr2[i]

            if fr2[2] > 1:
                s += fr2[2] * (fr2[2] - 1) // 2

            # i-au mai ramas numai valori prime impare
            for i in range(3, mx + 1, 2):
                if fr2[i] > 1:
                    s += fr2[i] * (fr2[i] - 1) // 2

            fout.write(str(s) + "\n")

if __name__ == "__main__":
    main()

</syntaxhighlight>