Cod sursa(job #3213964)

Utilizator David2007David Preda David2007 Data 13 martie 2024 17:18:50
Problema Medie Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.82 kb
#include <bits/stdc++.h>

using namespace std;

ifstream f("medie.in");
ofstream g("medie.out");

int n,sol,x;
int v[10000],fr[7005];

int main()
{
    f>>n;
    for(int i=1;i<=n;i++)
    {
        f>>v[i];
        fr[v[i]]++;
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=i+1;j<=n;j++)
        {
            if((v[i]+v[j])%2==0)
            {
                x=(v[i]+v[j])/2;
				if(v[i]==x && v[j]!=x)
                    fr[v[i]] -= 2;
				else if(v[j]==x && v[i]!=x)
                    fr[v[j]]-=2;
				else
                    fr[v[i]]-=2;

				sol+=fr[x];

				if (v[i]==x&&v[j]!=x)
                    fr[v[i]]+=2;
				else if(v[j]==x && v[i]!=x)
                    fr[v[j]]+=2;
				else
                    fr[v[i]]+=2;
            }
        }
    }
    g<<sol;
    return 0;
}