260C - Balls and Boxes - CodeForces Solution

#include <bits/stdc++.h>

using namespace std;

#define ll  long long
#define endl '\n'
#define cin_2d(vec, n, m) for(int i = 0; i < n; i++) for(int j = 0; j < m && cin >> vec[i][j]; j++);
#define fixed(n) fixed << setprecision(n)
#define cout_2d(vec, n, m) for(int i = 0; i < n; i++, cout << "\n") for(int j = 0; j < m && cout << vec[i][j] ; j++);
#define  loop(n)     for(int i=0; i<n; ++i)
#define  loopj(n)     for(int j=0; j<n; ++j)
#define  trav(a, b)   for(auto &a:b)
#define  YES         cout << "YES"<<endl;
#define  NO          cout << "NO" << endl;
#define yn cout<<(flag1?"YES":"NO")<<endl;
#define mem(a, b) memset(a , b , sizeof a);
#define FO ios_base::sync_with_stdio(false); cin.tie(NULL);
#define sz(s) (int)s.size()
#define all(vec) vec.begin(), vec.end()
#define debug(x) cout << #x << ": " << (x) << "\n";
#define debugLine(x) cout << #x << ": " << (x) << " ";
const ll OO = 0x3f3f3f3f, mod = 1e9 + 7, N = 1e3 + 5;
ll n, m, k, l, r;
bool flag1;


void _go() {
    cin >> n >> k;
    k--;
    vector<ll> v(n);
    loop(n)cin >> v[i];

    int val = *min_element(all(v));
    ll sum = 0;
    loop(n) {
        v[i] -= val;
        sum += val;
    }
    while (v[k])
    {
        v[k]--;
        k=(k-1+n)%n;
        sum++;
    }

    v[(k+n)%n]+=sum;
    loop(n)cout<<v[i]<<" ";
    cout<<endl;


}

int main() {
    //FO
//    freopen("input.txt", "r", stdin);
//    freopen("output.txt", "w", stdout);
    int t = 1;
    //cin >> t;
    while (t--) {
        _go();
    }
    return 0;
}