Cod sursa(job #1466118)

Utilizator cojocarugabiReality cojocarugabi Data 28 iulie 2015 17:48:41
Problema Puteri Scor 60
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.7 kb
# include <bits/stdc++.h>
# define md(x) md[x]
# define ll long long
using namespace std;
class Reader {
  public:
    Reader(const string& filename):
            m_stream(filename),
            m_pos(kBufferSize - 1),
            m_buffer(new char[kBufferSize]) {
        next();
    }

    Reader& operator>>(int& value) {
        value = 0;
        while (current() < '0' || current() > '9')
            next();
        while (current() >= '0' && current() <= '9') {
            value = value * 10 + current() - '0';
            next();
        }
        return *this;
    }

  private:
    const int kBufferSize = 32768;

    char current() {
        return m_buffer[m_pos];
    }

    void next() {
        if (!(++m_pos != kBufferSize)) {
            m_stream.read(m_buffer.get(), kBufferSize);
            m_pos = 0;
        }
    }

    ifstream m_stream;
    int m_pos;
    unique_ptr<char[]> m_buffer;
};
Reader fi("puteri.in");
ofstream fo("puteri.out");
const int pw[] = {-1,1};
const int nmax = 1 << 7;
int md[nmax + 777];
int ok(int p)
{
    bool k = 0;
    for (int i = 2;i*i <= p;++i)
    if (!(p%i))
    {
        p /= i;
        if (!(p%i)) return 0;
        k = 1 - k;
    }
    if (p > 1) k = 1 - k;
    return pw[k];
}
int main(void)
{
    int n;
    fi>>n;
    int * a;
    int * b;
    int * c;
    int ***s;
    a = new int [n+1];
    b = new int [n+1];
    c = new int [n+1];
    ll * ans;
    int nx = 0;
    for (int i = 1;i <= n;++i) fi>>a[i]>>b[i]>>c[i],nx = max(max(nx,a[i]),max(b[i],c[i]));
    nx *= 2;
    ans = new ll [nmax+5];
    s = new int** [nx];
    for (int i = 0;i < nx;++i) s[i] = new int*[nx];
    for (int i = 0;i < nx;++i)
        for (int j = 0;j < nx;++j) s[i][j] = new int[nx];
    for (int i = 0;i < nx;++i)
        for (int j = 0;j < nx;++j)
            memset(s[i][j],0,sizeof(s[i][j]));
    for (int mod = 2;mod <= nx;++mod)
    if (ok(mod))
    {
        for (int j = 0;j <= 555;++j) md[j] = j % mod;
        for (int i = 1;i <= n;++i)
            s[md(a[i])][md(b[i])][md(c[i])]++;
        for (int i = 1;i <= n;++i)
        if (s[md(a[i])][md(b[i])][md(c[i])])
        {
            int x = md(a[i]),y = md(b[i]),z = md(c[i]);
            int ix = md(mod - x),iy = md(mod - y),iz = md(mod - z);
            if (ix < 65 && iy < 65 && iz < 65)
            if (ix == x && iy == y && iz == z)
                ans[mod] += 1ll * (s[x][y][z] - 1) * s[x][y][z] / 2;
            else
                ans[mod] += 1ll * s[x][y][z] * s[ix][iy][iz];
            s[x][y][z] = s[ix][iy][iz] = 0;
        }
    }
    ll rs = 0;
    for (int i = 2;i <= nx;++i) rs += ok(i) * ans[i];
    return fo << rs << '\n',0;
}