Cod sursa(job #2003881)

Utilizator stefdascalescuStefan Dascalescu stefdascalescu Data 24 iulie 2017 12:21:41
Problema Medie Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.41 kb
#include<bits/stdc++.h>
using namespace std;
ifstream f("medie.in");
ofstream g("medie.out");
long long n,nr,fr[7001];
long long sol;
int main()
{
    f>>n;
    for(int i=1;i<=n;++i)
        f>>nr,fr[nr]++;
    for(int i=1;i<=7000;++i)
    {
        sol+=fr[i]*(fr[i]-1)*(fr[i]-2)/2;
        for(int j=1;j<i && i+j<=7000;++j)
            sol+=fr[i]*fr[i-j]*fr[i+j];
    }
    g<<sol;
    return 0;
}