[AtCoder] ABC 195 C – Comma

問題

方針

コンマの数が \( i \) 個となる数字 \( x \) の範囲は、\( 10^{3i} \leq x \leq 10^{3i + 3} – 1 \) なのでこの範囲を \( i \) を全探索すれば良いです。

コード

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

int main() {
    ll N;
    cin >> N;
    ll l = 1000;
    ll r = 1000000;
    ll ans = 0;
    for (ll i = 1; i <= 5; i++) {
        if (l <= N) {
            ans += i * (min(r - 1, N) - l + 1);
        } else {
            break;
        }
        l *= 1000;
        r *= 1000;
    }
    cout << ans << "\n";
    return 0;
}