Cod sursa(job #287172)

Utilizator ErgoVicol Sergiu Constantin Ergo Data 24 martie 2009 16:35:40
Problema Medie Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.65 kb
#include <fstream>
#define VMAX 7100
#define RVMAX 7000
#define NMAX 9005
#define ll long long
using namespace std;

ifstream fin("medie.in");
ofstream fout("medie.out");

ll V[VMAX], N, SMAX, vmax, NR[NMAX];

int main()
{
    ll i, a, j, k;
    fin >>N;
    for (i = 1; i <= N; i++)
    {
        fin >>a;
        if (a > vmax) vmax = a;
        V[a]++;
    }
    for (i = 1; i <= vmax; i++)
        if (V[i] > 0)
        {
            if (V[i] >= 3) SMAX += ( V[i]*(V[i]-1)*(V[i]-2) )/2  ;
            for (j = i+1, k = i-1; k >= 1 && j <= vmax; k--, j++)
                SMAX += V[k] * V[j] * V[i];
        }
    fout <<SMAX;
    fout.close();
    return 0;
}