Se consideră un șir format din n 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 n și c, și un șir de n numere naturale, se cere să se rezolve următoarele cerințe:

Se consideră un șir format din n 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 n și c, și un șir de n numere naturale, se cere să se rezolve următoarele cerințe:
1. dacă c = 1, atunci se cere să se afișeze lungimea maximă a unei secvențe neuniforme.
2. dacă c = 2, atunci se cere să se afișeze numărul total de secvențe neuniforme din șir.

Date de intrare

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

Date de ieșire

Dacă c = 1, atunci pe prima linie a fișierului numere.out va fi scris un singur număr ce reprezintă lungimea maximă a unei secvențe neuniforme. Dacă c = 2, atunci fișierul numere.out va conține un singur număr ce reprezintă numărul total de secvențe neuniforme.

Restricții și precizări

• 0 < n < 10.000
• 1 < valoare din șir < 10.000
• lungimea unei secvențe ≥ 2

Exemplul 1:

numere.in

6  1
14 2 49 3 35 1024  

numere.out

5

Explicație

Cele 6 numere sunt înlocuite cu valorile: 7, 2, 7, 3, 7, 2. Lungimea celei mai lungi secvențe neuniforme este 5; secvențele neuniforme cu acestă lungime sunt 7, 2, 7, 3, 7 sau 2, 7, 3, 7, 2.

Exemplul 2:

numere.in

10 2
14 8 3 25 6 24 20 1024 100 2

numere.out

9

Explicație

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

#include <bits/stdc++.h>
using namespace std;

int n , a[10001];

struct perechi{
    int p1 , p2 , val;
}f[10001];


ifstream cin("numere.in");
ofstream cout("numere.out");

bool prim(int x)
{
    int cnt = 0;
    for(int d = 1 ; d * d <= x ; ++d)
    {
        if(x % d == 0)
            cnt+=2;
        if(d * d == x)
            cnt--;
    }
    return cnt == 2;
}

int divi(int x)
{
    int max = 0;
    for(int d = 1 ; d * d <= x ; ++d)
    {
        if(x % d == 0)
        {
            if(prim(x/d) && x / d > max)
                max = x / d;
            if(prim(d) && d > max)
                max = d;
        }
    }
    return max;
}

int main()
{
    int c;
    cin >> n >> c;
    int x;
    for(int i = 1 ; i <= n ; ++i)
        cin >> x , a[i] = divi(x);
    for(int i = 1 ; i <= n ; ++i)
    {
        if(f[a[i]].val==0)
            f[a[i]].p1=i , f[a[i]].p2=i , f[a[i]].val++;
        else
            f[a[i]].p2=i , f[a[i]].val++;
    }
    if(c == 1)
    {
        int max = 0;
        for(int i = 1 ; i <= 10000 ; ++i)
            if(f[i].p2 - f[i].p1 + 1 > max)
                max = f[i].p2 - f[i].p1 + 1;
            cout << max << endl;
    }
    else
    {
        unsigned long long cnt = 0;
        for(int i = 1 ; i <= 10000 ; ++i)
            if(f[i].val > 1)
            cnt = cnt + f[i].val * (f[i].val-1) / 2;
        cout << cnt << endl;
    }
    return 0;
}