Codeforces Round #673 (Div. 2) D. Make Them Equal

    科技2022-07-10  104

    思路: 如果n不能整除于sum的话就无解。 那么考虑如何让每个数变成平均值,我们可以先把所有数都放到1的位置,再从1的地方分配给其他位置。 第一步:将所有数加到1的位置,我们枚举2-n,如果 a[i]%i=0的话直接操作,否则的话就先从1的位置拿一些数让i这个位置满足a[i]%i=0。考虑初始每个数都是正数,所以这一步不会使1的位置成为负数。 第二步:枚举 2-n,从1的位置拿sum/n个数过来。

    #pragma GCC optimize(2) #pragma GCC optimize(3) #include <bits/stdc++.h> using namespace std; typedef long long LL; const int N = 2e5 + 10; #define fi first #define se second #define pb push_back #define wzh(x) cerr<<#x<<'='<<x<<endl; int t,n,a[N]; int main() { ios::sync_with_stdio(false); for(cin>>t;t;t--){ cin>>n;int s=0; for(int i=1;i<=n;i++)cin>>a[i],s+=a[i]; if(s%n)cout<<-1<<'\n'; else{ vector<array<int,3> >v; for(int i=2;i<=n;i++){ if(a[i]%i==0){ v.pb({i,1,a[i]/i}); a[1]+=a[i]; a[i]=0; }else{ v.pb({1,i,i-a[i]%i}); v.pb({i,1,(a[i]+i-1)/i}); a[1]+=a[i]; a[i]=0; } } for(int i=2;i<=n;i++){ v.pb({1,i,s/n}); } cout<<v.size()<<'\n'; for(auto k:v)cout<<k[0]<<' '<<k[1]<<' '<<k[2]<<'\n'; } } return 0; }
    Processed: 0.010, SQL: 8