Cod sursa(job #2514767)

Utilizator LeVladzCiuperceanu Vlad LeVladz Data 26 decembrie 2019 19:26:51
Problema Medie Scor 60
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.68 kb
#include <fstream>

using namespace std;

ifstream fin("medie.in");
ofstream fout("medie.out");

int n,i,j,v[9005],f[7005],rest[7005];

int main()
{
    fin >> n;
    for (i=1; i<=n; i++)
    {
        fin >> v[i];
        f[v[i]]++; rest[v[i]] = v[i]%2;
    }
    long long sol = 0;
    for (i=1; i<n; i++)
        for (j=i+1; j<=n; j++)
            if (rest[v[i]] == rest[v[j]] && f[(v[i]+v[j])/2])
            {
                int val = (v[i]+v[j])/2; int ok = 0;
                if (v[i] == val)
                    ok++;
                if (v[j] == val)
                    ok++;
                sol += f[val]-ok;
            }
    fout << sol;
    return 0;
}