Codeforces Testing Round #10 B. Balancer

时间:2023-03-08 16:28:17
Codeforces Testing Round #10 B. Balancer

水题,只要遍历一遍,不够平均数的,从后面的借,比平均数多的,把多余的数添加到后面即可,注意数据范围

#include <iostream>
#include <vector>
#include <cmath>
using namespace std; int main(){
int n;
cin >> n;
vector<long long> a(n);
long long sum = ;
for(int i = ; i < n; ++ i){
cin >>a[i];
sum +=a[i];
}
sum/=n;
long long cnt = ;
for(int i = ; i < n-; ++ i){
a[i+] +=a[i]-sum;
cnt+=abs(a[i]-sum);
}
cout<<cnt<<endl;
}