Cod sursa(job #1694331)

Utilizator sucureiSucureiRobert sucurei Data 25 aprilie 2016 10:58:28
Problema Triplete Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.87 kb
#include <cstdio>
#include <algorithm>

#define DIM1 4100
#define DIM2 66000
#define BASE 32
using namespace std;

int N, M, X, Y, sol;
int V[DIM2], W[DIM2];
int A[DIM1][DIM1/BASE];

inline int ub(int X){
    int nr = 0;

    while(X != 0){
        nr ++;
        X = (X & (X - 1));
    }
    return nr;
}

int main(){

    freopen("triplete.in" ,"r", stdin );
    freopen("triplete.out","w", stdout);

    scanf("%d %d", &N, &M);

    for(int i = 1; i <= M; i ++){
        scanf("%d %d", &V[i], &W[i]);

        if(V[i] > W[i])
            swap(V[i], W[i]);

        A[V[i]][W[i] / BASE] += (1 << (W[i] % BASE));
    }

    for(int i = 1; i <= M; i ++){
        for(int j = 0; j <= N / BASE; j ++)
            sol += ub( A[ V[i] ][j] & A[ W[i] ][j] );
    }

    printf("%d\n", sol);

    fclose(stdin );
    fclose(stdout);

    return 0;
}