Cod sursa(job #2181151)

Utilizator EricEric Vilcu Eric Data 21 martie 2018 14:51:06
Problema Medie Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.45 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream f("medie.in");
ofstream g("medie.out");
int n,i,j,m,a[7003],S;
int main()
{
    f>>n;
    for(;n>0;--n)
    {
        f>>i;
        ++a[i];
    }
    S=0;
    for(m=1;m<=6;++m)
    {
        S+=a[m]*(a[m]-1)*(a[m]-2)/2;
        i=m-1;j=m+1;
        while(i>0&&j<7001)
        {
            S+=a[m]*a[i]*a[j];

            --i;++j;
        }

    }
    g<<S;
}