Cod sursa(job #2514776)

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

using namespace std;

FILE *fin = fopen("medie.in", "r");
FILE *fout = fopen("medie.out", "w");

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

int main()
{
    fscanf(fin, "%d", &n);
    for (i=1; i<=n; i++)
    {
        fscanf(fin, "%d", &v[i]);
        f[v[i]]++; rest[i] = v[i]%2;
    }
    long long sol = 0;
    for (i=1; i<n; ++i)
    {
        int x = rest[i];
        for (j=i+1; j<=n; ++j)
            if (x == rest[j])
            {
                int val = (v[i]+v[j])/2;
                int ok = 0;
                if (v[i] == val)
                    ok++;
                sol += f[val]-2*ok;
            }
    }
    fprintf(fout, "%lld", sol);
    return 0;
}