AtCoder Beginer Contest B問題でWAとなる理由がわかりません。

Question

AtCoder B問題でWAとなってしまう理由がわかりません。

前提・実現したいこと

AtCoder のAtCoder Beginer Contest083のB問題を解いていて、
"""
問題文
1以上 N以下の整数のうち、10進法での各桁の和が A以上 B以下であるものの総和を求めてください。
制約
1 ≤ N ≤ 10^4
1 ≤ A ≤ B ≤36
入力はすべて整数である

入力
入力は以下の形式で標準入力から与えられる。

N A B
出力
1以上 N以下の整数のうち、10進法での各桁の和が A以上 B以下であるものの総和を出力せよ。
"""
この問題を解いているとき、数列を文字列にして取り出して足していけばいいかなと思ったんですが、一部の数列にしか適用されず、その原因がわからず困っています。

発生している問題・エラーメッセージ

入力を(20 2 5)としたところ、
WA
正答84のところ50と出てしまいます。

該当のソースコード

line = list(map(int,input().split()))
n = line[0]
a = line[1]
b = line[2]

nums = []
sowa = 0
for num in range(n+1):
  count = len(str(num))
  for i in range(count-1):
    num = str(num)
    sowa += int(num[i])
  if a <= sowa and sowa <= b:
    nums.append(num)
answer = 0
for j in nums:
  answer += int(j)
print(answer)

試したこと

最後のprint(answer)をprint(nums)にしてリストの中身を確認したところ、
['11', '12', '13', '14']
でした。
20以下の整数のうち、各桁の和が 2以上 5以下なのは 2,3,4,5,11,12,13,14,20なのでこれらすべてが本来はリストに入っているはずです。

補足情報（FW/ツールのバージョンなど）

Answer 1

既に解決済みですが、ちょっとPython的にイケてない記述が散見されるので突っ込ませて下さい。

1

line = list(map(int,input().split()))
n = line[0]
a = line[1]
b = line[2]

アンパック代入を利用すれば一行で書け、しかもリストを経由しなくて済みます。

n, a, b = map(int, input().split())

2

sowa = 0
for num in range(n+1):
...

sowaは毎度リセットする必要があるのですから、for文の中に置くのが自然でしょう。

for num in range(n+1):
    sowa = 0
    ...

3

for num in range(n+1):
...
for i in range(count-1):
num = str(num)
...

numは外側のループだけに依存しますから、最奥のループで文字列化する必要は無い筈です。

for num in range(n+1):
  ...
  num = str(num)
  for i in range(count-1):
    ...

4

num = str(num)
for i in range(count):
  sowa += int(num[i])

numはこのとき文字列ですから、添え字を使わずとも巡回できます。

num = str(num)
for d in num:
  sowa += int(d)

5

if a <= sowa and sowa <= b:
nums.append(num)

Pythonでは次のように書くことができ、より直感的です。

if a <= sowa <= b:
  nums.append(num)

まとめ

以上の点を踏まえると、次のようにコードを書き換えることができます。
より簡潔に、分かり易くなったように思いませんか？

n, a, b = map(int, input().split())

nums = []
for num in range(n+1):
    num = str(num)
    sowa = 0
    
    for d in num:
        sowa += int(d)
        
    if a <= sowa <= b:
        nums.append(num)
        
answer = 0
for j in nums:
    answer += int(j)
    
print(answer)

発展

慣れてくれば次のように書くことも可能です。

n, a, b = map(int, input().split())

nums = [
    num
    for num in range(1, n+1)
    if a <= sum(int(d) for d in str(num)) <= b
]
print(sum(nums))

Comments

[sikotaro]

ご丁寧にありがとうございます。
とても勉強になりました。

Answer 2

  for i in range(count-1):

なぜここを1引いているのでしょう？これのせいで、1の位が参照できなくなっていませんか？
そうだとすれば、1桁の数はすべて飛ばされています。
また、sowaがリセットされていないので、コードの動作を追いかけると以下のようになると思います。

1. 前述の理由により、1桁はsowaの計算ができずにすべてスキップ。
2. 10の時に、十の位1がsowaに加算され、sowa=1。sowaが2未満のため、10はnumsに入らない。
3. 11の時に、十の位1がsowaに加算され、sowa=2。sowaが2以上のため、11がnumsに入る。
4. 12、13、14の時も十の位1が加算され、それぞれの時にsowaが3、4、5となり、それぞれnumsに入る。
5. 15以降はsowaが5を超えるため、numsに追加されない。

Comments

[sikotaro]

回答ありがとうございます。
今sowaをfor in の中に入れてcountの-1をなくしたところすべてリストに入りました！
countはlen関数で1から数えているので0から数えるrange関数に入れたらおかしくなるのではと思っていましたが、range(n)がn-1までしか数えないことを失念していました。
おかげさまで解決しました。ありがとうございました。