Cod sursa(job #1718285)

Utilizator andreicoman299Coman Andrei andreicoman299 Data 17 iunie 2016 11:44:21
Problema Triplete Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.32 kb
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <vector>
#include <math.h>

#define BUF_SIZE 16384
char buf[BUF_SIZE];
int pbuf=BUF_SIZE;
FILE*fi,*fo;
inline char nextch(){
    if(pbuf==BUF_SIZE){
        fread(buf, BUF_SIZE, 1, fi);
        pbuf=0;
    }
    return buf[pbuf++];
}
inline int nextnum(){
    int a=0;
    char c=nextch();
    while((c<'0' || c>'9') && c!='-')
        c=nextch();
    int semn=1;
    if(c=='-'){
        semn=-1;
        c=nextch();
    }
    while('0'<=c && c<='9'){
        a=a*10+c-'0';
        c=nextch();
    }
    return a*semn;
}

unsigned int v[4096][128];
int main(){
    fi=fopen("triplete.in","r");
    fo=fopen("triplete.out","w");
    int n=nextnum(), m=nextnum();
    for(int i=0;i<m;i++){
        int x=nextnum(), y=nextnum();
        x--;
        y--;
        v[x][y/32]=v[x][y/32]|(1<<(y%32));
        v[y][x/32]=v[y][x/32]|(1<<(x%32));
    }
    int s=0;
    for(int i=0;i<n;i++)
        for(int j=i+1;j<n;j++){
            if(v[i][j/32]&(1<<(j%32))){
                for(int k=0;k<128;k++)
                    s=s+__builtin_popcount(v[i][k]&v[j][k]);
            }
            //printf("%d %d %d\n", i, j, v[i][j/32]&(1<<(j%32)));
        }

    fprintf(fo,"%d", s/3);
    fclose(fi);
    fclose(fo);
    return 0;
}