Cod sursa(job #1473514)

Utilizator atatomirTatomir Alex atatomir Data 19 august 2015 15:44:28
Problema Medie Scor 60
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.84 kb
#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
#include <cmath>

using namespace std;

#define mp make_pair
#define pb push_back

#define maxN 9011
#define c v[i]
#define b v[j] //! time limit degeaba..

int n,i,j,a;
int v[maxN];
int cnt[maxN];
long long ans;

int main()
{
    freopen("medie.in","r",stdin);
    freopen("medie.out","w",stdout);

    scanf("%d",&n);
    for (i = 1; i <= n; i++){
        scanf("%d",&v[i]);
        cnt[ v[i] ]++;
    }

    for (i = 1; i <= n; i++){
        cnt[c]--;

        for (j = 1; j <= i-1; j++) {
            cnt[b]--;

            //! b+c == 2*a
            a = b+c;
            if (a % 2 == 0) {
                a >>= 1;
                ans += cnt[a];
            }

            cnt[b]++;
        }

        cnt[c]++;
    }

    printf("%lld",ans );

    return 0;
}