Cod sursa(job #3268593)

Utilizator AlexSerban21Serban Alexandru AlexSerban21 Data 16 ianuarie 2025 11:11:01
Problema Medie Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.59 kb
#include <fstream>
using namespace std;
ifstream fin ("medie.in");
ofstream fout ("medie.out");
int n,maxim,v[9001],fr[7001],sol[7001];
long long nrc;
int main()
{
    fin>>n;
    for (int i=1; i<=n; i++)
    {
        fin>>v[i];
        fr[v[i]]++;
        maxim=max (maxim,v[i]);
    }
    for (int i=1; i<=maxim; i++)
    {
        for (int j=i+1; j<=maxim; j++)
        {
            if ((i+j)%2==0)
                sol[(i+j)/2]+=fr[i]*fr[j];
        }
    }
    for (int i=1; i<=n; i++)
        nrc+=sol[v[i]]+1ll*(fr[v[i]]-1)*(fr[v[i]]-2)/2;
    fout<<nrc;
    return 0;
}