[C++]牛客 WY10 分苹果

最新推荐文章于 2025-01-26 23:13:27 发布

原创最新推荐文章于 2025-01-26 23:13:27 发布 · 311 阅读

0 ·

CC 4.0 BY-SA版权

牛客刷题专栏收录该内容

22 篇文章

订阅专栏

该博客探讨了一种算法问题，涉及如何通过移动苹果使得所有奶牛手中的苹果数量相等。输入描述为每只奶牛的苹果数量，算法需要找出在限制每次移动两个苹果的情况下，达到平分的最小操作次数。如果无法平分，则输出-1。示例展示了当奶牛数量为4，苹果总数为28636时，如何在3次移动后实现平分。博客内容涉及算法设计、遍历、条件判断和数值计算，适合对算法和编程感兴趣的读者。

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

输入描述：
每个输入包含一个测试用例。每个测试用例的第一行包含一个整数 n（1 <= n <= 100），接下来的一行包含 n 个整数 ai（1 <= ai <= 100）。

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

示例1
输入：
4
7 15 9 5
输出：
3

#include<iostream>
#include<vector>

using namespace std;

/*思路：
    1.先遍历数组，统计每个奶牛手中的苹果
    2.求平均数m，若可以整除，表示右分配方案，否则没有。
    3.遍历每个元素，比较当前元素与m得到差值，如果差值不是2的倍数，说明无论如何不存在移动方案使得
    条件成立。如果差值是2的倍数，就往后找大于平均值的元素作为填补。*/
int main(){
    int n;
    cin >> n;
    int sum = 0;
    vector<int> num(n, 0);
    for (int i = 0; i < n; i++) {
        cin >> num[i];
        sum += num[i];
    }
    if(sum % n != 0) 
    {
        cout << -1;
        return 0;
    }
    
    int avg = sum / n;
    int step = 0;
    for(int i = 0; i < n; i++) {
        int gap = num[i] - avg;
        if(gap == 0) continue;
        if (gap % 2 != 0) 
        {
            cout << -1;
            return 0;
        }
        
        for (int j = 0; j < n; j++) {
            if(i == j) continue;
            while (gap < 0 && num[j] > avg) {
                step++;
                num[i] += 2;
                num[j] -= 2;
                gap = num[i] - avg;
            }
        }
    }
    cout << step;
    return 0;
    
}