Cod sursa(job #2514778)

Utilizator LeVladzCiuperceanu Vlad LeVladz Data 26 decembrie 2019 19:43:52
Problema Medie Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.73 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],sol[9005];
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 solsol = 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;
                sol[val]++;
            }
    }
    for (i=1; i<=n; i++)
    {
        int x = v[i];
        solsol += sol[x]-f[x]+1;
    }
    fprintf(fout, "%lld", solsol);
    return 0;
}