Cod sursa(job #2102819)

Utilizator Bodo171Bogdan Pop Bodo171 Data 9 ianuarie 2018 15:17:00
Problema Puteri Scor 60
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.22 kb
#include <iostream>
#include <fstream>
using namespace std;
const int nmax=100005;
const int nm=128;
int ap[128][128][128],lst[128][128][128];
int a[nmax],b[nmax],c[nmax],d[200],e[200],inv[200];
long long pairs,tot;
int n,i,j,mod,x,y,z;
int main()
{
    ifstream f("puteri.in");
    ofstream g("puteri.out");
    f>>n;
    for(i=1;i<=n;i++)
    {
        f>>a[i]>>b[i]>>c[i];
    }
    for(i=2;i<=nm;i++)
        if(d[i]==0)
    {
        for(j=i;j<=nm;j+=i)
        {
            if(d[j]!=-1)d[j]++;
            if(j%(i*i)==0) d[j]=-1;
        }
    }
    for(mod=2;mod<=nm;mod++)
        if(d[mod]!=-1)
    {
        pairs=0;
        for(i=1;i<=nm;i++)
        {
            e[i]=e[i-1]+1;
            if(e[i]==mod)
                e[i]=0;
        }
        for(i=1;i<mod;i++)
            inv[i]=mod-i;
        for(i=1;i<=n;i++)
        {
            x=e[a[i]];y=e[b[i]];z=e[c[i]];
            if(lst[inv[x]][inv[y]][inv[z]]==mod)
              pairs+=1LL*ap[inv[x]][inv[y]][inv[z]];
            if(lst[x][y][z]==mod)ap[x][y][z]++;
            else ap[x][y][z]=1,lst[x][y][z]=mod;
        }
        if(d[mod]%2) tot+=1LL*pairs;
        else tot-=1LL*pairs;
    }
    g<<tot;
    return 0;
}