【AtCoder】ABC 444 B - Digit Sum

2026-02-08

AtCoder/ABC/B問題 AtCoder/ABC/200点問題 AtCoder/灰Diff AtCoder/NoviSteps/未分類 AtCoder/数学/各桁の和競技プログラミング

B - Digit Sum

AtCoder is a programming contest site for anyone from beginners to experts. We hold weekly programming contests online.

AtCoder

実行時間制限: 2 sec / メモリ制限: 1024 MiB / Difficulty: 36 / NoviSteps: ??? / 配点: 200 点

問題概要

$N$ 以下の正整数のうち、各桁の和が $K$ であるものの個数を求めよ。

制約

$1\leq N,K \leq 10^5$
入力は全て整数

考察

$x = 1, 2, \cdots, N$ について、各桁の和を求め、それが $K$ であれば答えのカウンターを増やしていけばよい。

$x$ の各桁の和を求める処理は、別に関数化しておくと便利であろう。

実装例

1.#include <bits/stdc++.h>
2.using namespace std;
3. 
4.#define rep(i, start, end) for (auto i = (start); (i) < (end); (i)++)
5. 
6.// ======================================== //
7. 
8.int main()
9.{
10.    int N, K;
11.    cin >> N >> K;
12. 
13.    auto digit_sum = [&](int x) -> int
14.    {
15.        int res = 0;
16.        while (x > 0)
17.        {
18.            res += x % 10;
19.            x /= 10;
20.        }
21.        return res;
22.    };
23. 
24.    int ans = 0;
25.    rep(x, 1, N + 1)
26.    {
27.        if (digit_sum(x) == K)
28.            ans++;
29.    }
30. 
31.    cout << ans << endl;
32. 
33.    return 0;
34.}