Cod sursa(job #1976728)

Utilizator RaresEGaySopterean Adrian RaresEGay Data 4 mai 2017 08:53:02
Problema Medie Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.58 kb
#include <fstream>
#include <iostream>

#define MAXN 9005
#define NR 7005

using namespace std;

ifstream f ("medie.in");
ofstream g ("medie.out");

int n;
long long sol;
int v[MAXN], s[NR];

int main(){
    f >> n;

    for(int i = 1; i <= n; ++i){
        f >> v[i];
        s[v[i]]++;
    }
    for(int i = 1; i < n; ++i){
        for(int j = i + 1; j <= n; ++j){
            int k = v[j] + v[i];
            if(k % 2 == 0 && s[k/2]){
                sol += s[k/2] - 2;
            }
        }
    }
    if(sol < 0) g << sol * -1 << '\n';
    else g << sol;
}