Filya and Homework
CodeForces - 714BToday, hedgehog Filya went to school for the very first time! Teacher gave him a homework which Filya was unable to complete without your help.
Filya is given an array of non-negative integers a1, a2, ..., an. First, he pick an integer x and then he adds x to some elements of the array (no more than once), subtract x from some other elements (also, no more than once) and do no change other elements. He wants all elements of the array to be equal.
Now he wonders if it's possible to pick such integer x and change some elements of the array using this x in order to make all elements equal.
The first line of the input contains an integer n (1 ≤ n ≤ 100 000) — the number of integers in the Filya's array. The second line contains n integers a1, a2, ..., an (0 ≤ ai ≤ 109) — elements of the array.
If it's impossible to make all elements of the array equal using the process given in the problem statement, then print "NO" (without quotes) in the only line of the output. Otherwise print "YES" (without quotes).
5 1 3 3 2 1
YES
5 1 2 3 4 5
NO
In the first sample Filya should select x = 1, then add it to the first and the last elements of the array and subtract from the second and the third elements.
code:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <map>
#include <algorithm>
using namespace std;
const int MAXN = 1e5+100;
int a[MAXN];
int n;
map<int,int>mp;
int main(){
int i;
int cnt = 0;
int b[4];
scanf("%d",&n);
for(i = 0; i < n; i++){
scanf("%d",&a[i]);
if(!mp.count(a[i]))
mp[a[i]] = cnt++;
}
if(cnt>3)
printf("NO\n");
else{
if(cnt <= 2)
printf("YES\n");
else{
cnt = 0;
map<int,int>::iterator it;
for(it = mp.begin(); it != mp.end(); it++)
b[cnt++] = (*it).first;
sort(b,b+cnt);
if(b[2]-b[1] != b[1]-b[0])
printf("NO\n");
else
printf("YES\n");
}
}
return 0;
}
刺猬Filya首次上学并获得了一项数学家庭作业,任务是通过加减某个整数x使数组中的所有元素相等。本题探讨了实现这一目标的可能性,并提供了解决方案。
246

被折叠的 条评论
为什么被折叠?



