Cod sursa(job #60150)

Utilizator cos_minBondane Cosmin cos_min Data 12 mai 2007 19:14:10
Problema Medie Scor 40
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.85 kb
#include <stdio.h>
#include <ext/hash_map>
using namespace std;
using namespace __gnu_cxx;

#define in "medie.in"
#define out "medie.out"
#define dim 9001

hash_map<int,int> H1;
hash_map<int,int> H2;
int A[dim], N;
int D[dim];

int main()
{
    freopen(in,"r",stdin);
    freopen(out,"w",stdout);
    
    scanf("%d", &N);
    
    for ( int i = 1; i <= N; i++ )
    {
        scanf("%d", &A[i]);
        H2[A[i]]++;
    }
    
    int med;
    int total = 0;
        
    for ( int i = 1; i < N; i++ )
    {
        for ( int j = i+1; j <= N; j++ )
        {
            med = A[i] + A[j];
            if ( med&1 ) continue;
            H1[med/2]++; 
        }
    } 
    
    for ( int k = 1; k <= N; k++ )
        total += (H1[A[k]]>0)*H1[A[k]] - (H1[A[k]]>0)*H2[A[k]] + (H1[A[k]]>0)*1;
    
    printf("%d", total);
}