Cod sursa(job #3031233)

Utilizator matwudemagogul matwu Data 19 martie 2023 09:20:11
Problema Medie Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.54 kb
#include <bits/stdc++.h>
using namespace std;
const int N = 9e3 + 1;
int f1[(int)7e3 + 1], f2[(int)7e3 + 1];
int n;
vector<int> v(N);
int main(){
	ifstream fin("medie.in");
	ofstream fout("medie.out");
	fin.tie(0)->sync_with_stdio(0);
	fin >> n;
	for(int i = 1; i <= n; i++){
		fin >> v[i];
		f1[v[i]]++;
		for(int j = 1; j < i; j++){
			if((v[j] + v[i]) % 2 == 0){
				f2[(v[j] + v[i]) / 2]++;
			}
		}
	}
	long long ans = 0;
	for(int i = 1; i <= n; i++){
		int nr = f2[v[i]] - f1[v[i]] + 1;
		ans += nr;
	}
	fout << ans;
}