[AtCoder] ABC 192 C – Kaprekar Number

問題

方針

シミュレーションします。

コード

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

ll func(ll x) {
    vector<ll> v;
    while (x > 0) {
        v.push_back(x % 10);
        x /= 10;
    }
    sort(v.begin(), v.end());
    ll ret = 0;
    ll d = 1;
    for (int i = 0; i < v.size(); i++) {
        ret += d * (v[i] - v[v.size() - i - 1]);
        d *= 10;
    }
    return ret;
}

int main() {
    ll N, K;
    cin >> N >> K;
    ll ans = N;
    for (int i = 0; i < K; i++) {
        ans = func(ans);
    }
    cout << ans << "\n";
    return 0;
}