Cod sursa(job #3304400)

Utilizator unomMirel Costel unom Data 23 iulie 2025 11:20:20
Problema Puteri Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.21 kb
#include <fstream>

using namespace std;

struct el
{
    int a, b, c;
};

ifstream in("puteri.in");
ofstream out("puteri.out");
int n;
long long ans;
el v[100005];
int frec[130][130][130];
int mobius[205];
int rest[130][130];

void build_mobius()
{
    mobius[1] = -1;

    for(int i = 1; i<=128; i++)
    {
        if(mobius[i])
        {
            mobius[i] = -mobius[i];

            for(int j = 2 * i; j<=128; j += i)
            {
                mobius[j] += mobius[i];
            }
        }
    }

    for(int i = 1; i<=128; i++)
    {
        mobius[i] *= -1;
    }
}

int main()
{
    in>>n;

    build_mobius();

    for(int i = 0; i<=128; i++)
    {
        for(int j = 2; j<=128; j++)
        {
            rest[i][j] = i % j;
        }
    }

    for(int i = 1; i<=n; i++)
    {
        in>>v[i].a>>v[i].b>>v[i].c;
    }

    for(int i = 2; i<=128; i++)
    {
        for(int j = 1; j<=n; j++)
        {
            frec[rest[v[j].a][i]][rest[v[j].b][i]][rest[v[j].c][i]]++;
        }

        long long rez = 0;
        for(int a = 0; a<=min(64, i - 1); a++)
        {
            for(int b = 0; b<=min(64, i - 1); b++)
            {
                for(int c = 0; c<=min(64, i - 1); c++)
                {
                    int ra = i - a;
                    int rb = i - b;
                    int rc = i - c;

                    if(ra == i) ra = 0;
                    if(rb == i) rb = 0;
                    if(rc == i) rc = 0;

                    if(a == ra && b == rb && c == rc)
                    {
                        rez += max(0, (frec[a][b][c] * (frec[a][b][c] - 1)) / 2);
                    }
                    else
                    {
                        rez += (frec[a][b][c] * frec[ra][rb][rc]) / 2;

                        //out<<a<<" "<<b<<" "<<c<<" "<<frec[a][b][c] * frec[ra][rb][rc]<<'\n';
                    }
                }
            }
        }

        ans += rez * mobius[i];

        //out<<i<<" -> "<<rez<<'\n';

        for(int j = 1; j<=n; j++)
        {
            frec[rest[v[j].a][i]][rest[v[j].b][i]][rest[v[j].c][i]]--;
        }

    }

    out<<ans;

    return 0;
}