Cod sursa(job #2876247)

Utilizator BlueLuca888Girbovan Robert Luca BlueLuca888 Data 23 martie 2022 10:16:47
Problema Medie Scor 100
Compilator cpp-64 Status done
Runda masonii Marime 0.65 kb
#include <bits/stdc++.h>
#pragma GCC optimize ("Ofast")

using namespace std;

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

const int LIM = 7000;
int n, x, freq[LIM + 5];
long long sol;

int main (){
    ios_base::sync_with_stdio(false);
    fin.tie(nullptr);
    fout.tie(nullptr);

    fin>>n;
    for(int i=1; i<=n; i++){
        fin>>x;
        freq[x]++;
    }

    for(int i=1; i<=LIM; i++){
        sol += (long long)freq[i] * (freq[i]-1) * (freq[i]-2) / 2;
        for(int dif=1; 1<=i-dif && i+dif<=LIM; dif++)
            sol += (long long)freq[i] * freq[i-dif] * freq[i+dif];
    }
    fout<<sol;
    return 0;
}