网易编程题——分苹果

我怕爱的太早我们不能终老 提交于 2020-01-11 03:06:33

题目描述
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;

int main() {
    int n; 
    cin >> n;
    vector<int> nums;
    int total = 0;
    while (n--) {
        int tmp;
        cin >> tmp;
        total += tmp;
        nums.push_back(tmp);
    }
    int count = 0;
    // 不能均分
    if (total % nums.size() != 0) {
        cout << -1 << endl;
        return 0;
    }
    else {
        int average = total / nums.size();
        // 偶数 + 偶数(2) = 偶数
        if (average % 2 == 0) {
            for (int i : nums) {
                if (i % 2 != 0) {
                    cout << -1 << endl;
                    return 0;
                }
            }
        } else {
            // 奇数 + 偶数(2) = 奇数
            for (int i : nums) {
                if (i % 2 == 0) {
                    cout << -1 << endl;
                    return 0;
                }
            }
        }
        for (int i : nums) {
            if (i > average) {
                count += ((i - average) / 2);
            }
        }
    }
    cout << count << endl;
    return 0;
}

我发现经常写算法,头脑就会保持比较清醒的状态,写起代码来,思路也比较清晰,敲键盘也比较灵活。
写的多了,也就熟能生巧了,所以应该多写,多积累。

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!