Cod sursa(job #2223271)

Utilizator shantih1Alex S Hill shantih1 Data 19 iulie 2018 16:26:43
Problema Medie Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.43 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];
bool b[9000];
long long rez;

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