Cod sursa(job #2543892)

Utilizator PatrickCplusplusPatrick Kristian Ondreovici PatrickCplusplus Data 11 februarie 2020 16:59:34
Problema Medie Scor 60
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.92 kb
#include <bits/stdc++.h>

using namespace std;

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

int n, v[9005], fr[7005], fr2[7005], fr3[7005];
long long ans;


int main()
{
    fin >> n;
    for (int i = 1; i <= n; ++i)
        fin >> v[i];
    for (int i = 1; i < n; ++i)
    {
        for (int j = i + 1; j <= n; ++j)
        {
            if (j - i > 1) fr2[v[j - 1]]++;
            if (v[i] % 2 != v[j] % 2) continue;
            int x = (v[i] + v[j]) / 2;
            ans += fr[x] + fr2[x];
        }
        for (int j = i + 1; j < n; ++j) fr2[v[j]] = 0;
        fr[v[i]]++;
    }
    for (int i = n; i > 1; --i)
    {
        for (int j = i - 1; j >= 1; --j)
        {
            if (v[i] % 2 != v[j] % 2) continue;
            int x = (v[i] + v[j]) / 2;
            ans += fr3[x];
        }
        fr3[v[i]]++;
    }
    fout << ans;
    fin.close();
    fout.close();
    return 0;
}