Mai intai trebuie sa te autentifici.
Cod sursa(job #2324208)
Utilizator | Data | 20 ianuarie 2019 13:37:46 | |
---|---|---|---|
Problema | Medie | Scor | 100 |
Compilator | cpp-64 | Status | done |
Runda | Arhiva de probleme | Marime | 0.69 kb |
#include <fstream>
#include <algorithm>
using namespace std;
ifstream fin("medie.in");
ofstream fout("medie.out");
int N, vf[7005];
long long sol;
int main()
{
fin >> N;
int x;
for(int i = 1; i <= N; i++)
{
fin >> x;
vf[x]++;
}
for(int i = 1; i <= 7000; i++)
if(vf[i] >= 1)
{
if(vf[i] >= 3)
sol += (vf[i]) * (vf[i] - 1) * (vf[i] - 2) / 2;
for(int j = 1; i - j >= 1 && i + j <= 7000; j++)
if(vf[i - j] >= 1 && vf[i + j] >= 1)
sol += vf[i] * vf[i - j] * vf[i + j];
}
fout << sol << '\n';
return 0;
}