Cod sursa(job #2759828)

Utilizator MarcGrecMarc Grec MarcGrec Data 20 iunie 2021 19:29:50
Problema Medie Scor 20
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.81 kb
#define MAX_V 7000
#define MIN_V 1
#define MUL(a, b) (((int64_t) (a)) * ((int64_t) (b)))

#include <iostream>
#include <fstream>
#include <algorithm>
#include <cstdint>
using namespace std;

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

int F[MAX_V + 1];

int main()
{
    int n;
    fin >> n;
    
    for (int i = 1; i <= n; ++i)
    {
        int x;
        fin >> x;
        
        ++F[x];
    }
    
    int64_t res = 0;
    
    for (int i = MIN_V; i <= MAX_V; ++i)
    {
        res += MUL(MUL(F[i], F[i] - 1), F[i] - 2) >> 1;
        
        for (int d = 1; ((i - d) >= MIN_V) && ((i + d) <= MAX_V); ++d)
        {
            res += min(min(F[i - d], F[i]), F[i + d]);
        }
    }
    
    fout << res;
    
    fin.close();
    fout.close();
    return 0;
}