Cod sursa(job #1333628)

Utilizator chiriacandrei25Chiriac Andrei chiriacandrei25 Data 3 februarie 2015 13:57:48
Problema Puteri Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.97 kb
#include <cstdio>
#include <vector>
#define MOD 100019
#define Nmax 100005

using namespace std;

struct triplet
{
    int a,b,c,cnt;
    bool operator == (const triplet A) const
    {
        return (a==A.a && b==A.b && c==A.c);
    }
} v[Nmax],aux[Nmax];
vector <triplet> H[MOD];
int ciur[130],n;

inline void AH(triplet w)
{
    int key=(100*w.a+10*w.b+w.c)%MOD;
    vector <triplet> ::iterator it;
    for(it=H[key].begin();it!=H[key].end();++it)
        if(*it==w)
        {
            it->cnt++; return;
        }
    w.cnt=1;
    H[key].push_back(w);
}

inline void Delete(triplet w)
{
    int key=(100*w.a+10*w.b+w.c)%MOD;
    vector <triplet> ::iterator it;
    for(it=H[key].begin();it!=H[key].end();++it)
        if(*it==w)
        {
            H[key].erase(it); return;
        }
}

inline int SH(triplet w)
{
    int key=(100*w.a+10*w.b+w.c)%MOD;
    vector <triplet> ::iterator it;
    for(it=H[key].begin();it!=H[key].end();++it)
        if(*it==w) return it->cnt;
    return 0;
}

inline long long Count(int p)
{
    triplet w;
    int i;
    long long sol=0;
    for(i=1;i<=n;++i)
    {
        w.a=v[i].a%p; w.b=v[i].b%p; w.c=v[i].c%p; AH(w);
        aux[i]=w;
    }
    for(i=1;i<=n;++i)
    {
        w.a=(3*p-aux[i].a)%p; w.b=(3*p-aux[i].b)%p; w.c=(3*p-aux[i].c)%p;
        sol+=SH(w);
        if(w==aux[i]) --sol;
    }
    for(i=1;i<=n;++i) Delete(aux[i]);
    return sol/2;
}

int main()
{
    long long ans=0;
    int i,j;
    freopen ("puteri.in","r",stdin);
    freopen ("puteri.out","w",stdout);
    scanf("%d", &n);
    for(i=1;i<=n;++i) scanf("%d%d%d", &v[i].a,&v[i].b,&v[i].c);
    for(i=2;i<=128;++i)
        if(!ciur[i])
        {
            for(j=i*2;j<=128;j+=i) ++ciur[j];
            ans+=Count(i);
        }
        else
        {
            for(j=i*2;j<=128;j+=i) ciur[j]-=(ciur[i]-1);
            ans-=Count(i)*(ciur[i]-1);
        }
    printf("%lld\n", ans);
    return 0;
}