分苹果
问题描述
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; }
|