Cod sursa(job #1881671)

Utilizator AlexTheDagonBogdan Tudor AlexTheDagon Data 16 februarie 2017 17:35:12
Problema Medie Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.62 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <algorithm>
#define pb push_back
using namespace std;
ifstream in("medie.in");
ofstream out("medie.out");
int x,fr[15005],n;
vector <int> v;
long long nr,rez;
int main()
{
    in>>n;
    for(int i=1;i<=n;++i)
    {
        in>>x;
        if(fr[x]==0)v.pb(x);
        ++fr[x];
    }
    sort(v.begin(),v.end());
    for(int i=0;i<v.size();++i)
    {
        for(int j=0;j<i;++j)
        {
            nr+=fr[v[i]]*fr[v[j]]*fr[2*v[i]-v[j]];
        }
        nr+=(fr[v[i]]*(fr[v[i]]-1)*(fr[v[i]]-2)/2);
    }
    out<<nr;
    return 0;
}