Cod sursa(job #2876333)

Utilizator AlexNicuNicu Alexandru AlexNicu Data 23 martie 2022 10:51:24
Problema Medie Scor 100
Compilator cpp-64 Status done
Runda masonii Marime 0.52 kb
#include <fstream>

using namespace std;

ifstream cin ( "medie.in" );
ofstream cout ( "medie.out" );

#define VMAX 7000

int f[VMAX + 1];

int main() {
    long long n, i, ans, x, j, k;
    cin >> n;
    for ( i = 0; i < n; i++ ) {
        cin >> x;
        f[x]++;
    }
    ans = 0;
    for ( i = 1; i <= VMAX; i++ ) {
        ans += f[i] * ( f[i] - 1 ) * ( f[i] - 2 ) / 2;
        for ( j = i - 1, k = i + 1; j >= 1 && k <= VMAX; j--, k++ ) {
            ans += f[i] * f[j] * f[k];
        }
    }
    cout << ans;
    return 0;
}