Cod sursa(job #2223269)

Utilizator shantih1Alex S Hill shantih1 Data 19 iulie 2018 16:22:46
Problema Medie Scor 60
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.4 kb
#include <iostream>
#include <fstream>

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

int n,s,i,j,p,v[9005],f[7005];
long long rez;

int main() {
	
	fin>>n;
	for(i=1;i<=n;i++)
	{
		fin>>v[i];
		f[v[i]]++;
	}
	
	for(i=1;i<n;i++)
	{
		p=v[i]%2;
		for(j=i+1;j<=n;j++)
			if(v[j]%2==p)
			{
				s=(v[i]+v[j])/2;
				f[v[i]]-=2;
				rez+=f[s];
				f[v[i]]+=2;
			}
	}
	fout<<rez<<"\n";
}