Cod sursa(job #2737219)

Utilizator George_CristianGeorge Dan-Cristian George_Cristian Data 4 aprilie 2021 15:38:45
Problema Medie Scor 60
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.67 kb
#include <fstream>

using namespace std;

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

int n, v[7005], frecvrez[7005], frecv[7005];

void citire() {
    f >> n;
    for (int i = 0; i < n; ++i) {
        f >> v[i];
        frecv[v[i]]++;
    }
}

void creare_frecvrez() {
    for (int i = 0; i < n; ++i)
        for (int j = i + 1; j < n; ++j)
            if (v[i] % 2 == v[j] % 2)
                frecvrez[(v[i] + v[j]) / 2]++;
}

void afisare() {
    long long sol = 0;
    for (int i = 0; i < n; ++i)
        if (frecvrez[v[i]])
            sol += frecvrez[v[i]] - frecv[v[i]] + 1;
    g << sol;
}

int main() {
    citire();
    creare_frecvrez();
    afisare();
    return 0;
}