Cod sursa(job #2406028)

Utilizator Asgari_ArminArmin Asgari Asgari_Armin Data 15 aprilie 2019 12:28:56
Problema Puteri Scor 50
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.95 kb
#include <cstdio>

const int prime[32] = {2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47, 53, 59, 61, 67, 71, 73, 79, 83, 89, 97, 101, 103, 107, 109, 113, 127, 128};
int a[100001], b[100001], c[100001], n, d[128][128][128], nr, s = 1;
long long v[128], sol;

void bkt(int k) {
    if(k == 31) {
        int i, j, l, v[65];
        long long temp;
        temp = 0;
        if(nr == 0)
            return;
        for(i = 0; i <= 64; ++i) {
            v[i] = i % s;
        }
        for(i = 1; i <= n; ++i) {
            ++d[v[a[i]]][v[b[i]]][v[c[i]]];
        }
        for(i = 0; i < s; ++i) {
            for(j = 0; j < s; ++j) {
                for(l = 0; l <= s / 2; ++l) {
                    int newi = (s - i), newj = s - j, newl = s - l;
                    if(newi == s)newi = 0;
                    if(newj == s)newj = 0;
                    if(newl == s) newl = 0;
                    if(i == newi && j == newj && l == newl) {
                        temp += (long long) d[i][j][l] * (d[i][j][l] - 1) / 2;
                        d[i][j][l] = 0;
                    } else {
                        temp += (long long) d[i][j][l] * d[newi][newj][newl];
                        d[i][j][l] = d[newi][newj][newl] = 0;
                    }
                }
            }
        }
        //printf("%d\n", temp);
        if(nr % 2 == 0) {
            sol -= temp;
        } else {
            sol += temp;
        }
    } else {
        bkt(k + 1);
        ++nr;
        s *= prime[k];
        if(s < 128)
            bkt(k + 1);
        --nr;
        s /= prime[k];
    }
}

int main() {
    FILE *fin, *fout;
    fin = fopen( "puteri.in", "r" );
    fout = fopen( "puteri.out", "w" );
    int i;
    fscanf(fin, "%d", &n);
    for(i = 1; i <= n; ++i)
        fscanf(fin, "%d %d %d", &a[i], &b[i], &c[i]);
    bkt(0);
    fprintf(fout, "%lld\n", sol);
    fclose( fin );
    fclose( fout );
    return 0;
}