0%

网易2017招聘第三题

分苹果

问题描述

n 只奶牛坐在一排,每个奶牛拥有 ai 个苹果,现在你要在它们之间转移苹果,使得最后所有奶牛拥有的苹果数都相同,每一次,你只能从一只奶牛身上拿走恰好两个苹果到另一个奶牛上,问最少需要移动多少次可以平分苹果,如果方案不存在输出 -1。

输入描述:

每个输入包含一个测试用例。每个测试用例的第一行包含一个整数 n(1 <= n <= 100),接下来的一行包含 n 个整数 ai(1 <= ai <= 100)。

输出描述:

输出一行表示最少需要移动多少次可以平分苹果,如果方案不存在则输出 -1。

解法分析

就是首先检查一下能不能均分。就是和要是总数的倍数。
如果不行则输出-1.
如果可以呢
遍历数组
如果数组中的数减掉均值若不可以整除2,则输出-1。
如果都可以,那么输出差大于2的数组元素除以2的总数。

if((data[i] - mean) % 2 != 0 && (data[i] - mean) >=2 )
{
count += (data[i] - mean) / 2;
}
我之前用的是其他的算法,但是没有这个方法好,这里只采用这种方法。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int main() {
int n;
std::cin >> n;
vector<int> data;
for (int i =0; i < n; i++)
{
int temp;
cin >> temp;
data.push_back(temp);
}
int sum = 0;
for (int i = 0; i < n; i++)
{
sum += data[i];
}
if(sum % n != 0)
{
cout << -1;
return 0;
}
int count = 0;
int mean = sum / n;
for(int i = 0; i < n; i++)
{
int minus = data[i] - mean;
if(minus % 2 != 0)
{
cout << -1;
return 0;
}
else
{
if(minus >= 0)
count += minus / 2;
}
}
cout << count;

return 0;
}