C++ 雪人部落储备粮分配问题

C++ 代码

#include <iostream>
#include <vector>
using namespace std;

int main() {
    int n;
    cin >> n;
    vector<int> foods(n);
    long long sum = 0;
    for (int i = 0; i < n; i++) {
        cin >> foods[i];
        sum += foods[i];
    }
    long long avg = sum / n;
    long long extra = 0;
    for (int i = 0; i < n; i++) {
        if (foods[i] < avg) {
            extra += avg - foods[i];
        }
    }
    cout << extra << endl;
    return 0;
}