Cod sursa(job #2679660)

Utilizator gasparrobert95Gaspar Robert Andrei gasparrobert95 Data 1 decembrie 2020 10:11:46
Problema Medie Scor 60
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.67 kb
#include <bits/stdc++.h>
#define ll long long
using namespace std;
ifstream fin("medie.in");
ofstream fout("medie.out");
ll n, rez, v[9005], ap[7005];

int main() {
    fin >> n;
    for (int i = 1; i <= n; ++i) {
        fin >> v[i];
        ++ap[v[i]];
    }
    for (int i = 1; i <= n; ++i) {
        for (int j = i + 1; j <= n; ++j) {
            if ((v[i] + v[j]) % 2 != 0)
                continue;
            if (ap[(v[i] + v[j]) / 2] >= 1 && v[j] != v[i])
                rez += ap[(v[i] + v[j]) / 2];
            else if (ap[(v[i] + v[j]) / 2] >= 3)
                rez += ap[(v[i] + v[j]) / 2] - 2;
        }
    }
    fout << rez;
    return 0;
}