AtCoderの過去問ABC165のA問題の不正解の理由

AtCoderのABC165のA問題について質問させてください。
https://atcoder.jp/contests/abc165/tasks/abc165_a

下記で提出したのですが、不正解でしたが理由が分かりません。

python
1k = int(input())
2a, b = list(map(int, input().split()))
3s = [i for i in range(a, b) if i % k == 0]
4if a == b and a % k == 0:
5	print('OK')  
6elif len(s)==0:
7	print('NG')	
8else:
9	print('OK')
10

実装で無理やり感は否めないですが、自分なりのコードテストはOKでした。
解説サイトなどで別解は載っているのですが、自分の提出で何が駄目だったのか恥ずかしながら,腑に落ちず、
お分かりになられる方いらっしゃればと思いまして質問させて頂きました。

行動規範の内容に同意します

回答2件

ベストアンサー

k,a,b = 2,1,2の場合、正解はOKですがNGになってしまいます。
原因はrange(a,b)はa～b-1の間の値しかとらないためです。

投稿2020/07/05 08:06

can110

総合スコア38266

m8aFm77kKU9JuID

2020/07/05 09:22

なるほど、ありがとうございました。

行動規範の内容に同意します

つぎのプログラムを研究してみてください。
質問文のコードでは、 TEST にある例で失敗することがあります。
k, a, b が 1..10 のあいだで試すと、他にも失敗する例ｆがあります。

solve2() とした判定方法では 1..200 の間で失敗することはないです。
(時間がかかるから試していませんが、 1..1000 でも失敗することはないと思います)

p.py

python3
1def solve(k, a, b):
2  s = [i for i in range(a, b) if i % k == 0]
3  if a == b and a % k == 0:
4    return "OK"  
5  elif len(s) == 0:
6    return "NG"   
7  else:
8    return "OK"
9
10def solve2(k, a, b):
11  if a % k == 0:
12    return "OK"
13
14  if (a // k + 1) * k <= b:
15    return "OK"
16  return "NG"
17
18def check(k, a, b):
19  for i in range(a, b + 1):
20    if i % k == 0:
21      return "OK"
22  return "NG"
23
24TEST = [
25  [[7, 500, 600], "OK"],
26  [[4, 5, 7], "NG"],
27  [[1, 11, 11], "OK"],
28  [[2, 11, 11], "NG"],
29  
30  [[11, 11, 11], "OK"],
31  [[12, 11, 12], "OK"]
32]
33for t in TEST:
34  k, a, b = t[0]
35  if t[1] != solve(k, a, b):
36    print("Fail ", t)
37
38print("-----------------")
39print("solve")
40for k in range(1, 5):
41  for a in range(1, 5):
42    for b in range(a, 5):
43      if check(k, a, b) != solve(k, a, b):
44        print("Fail", k, a, b)
45
46N = 200 # 1000
47print("-----------------")
48print("solve2")
49for k in range(1, N + 1):
50  for a in range(1, N + 1):
51    for b in range(a, N + 1):
52      if check(k, a, b) != solve2(k, a, b):
53        print("Fail", k, a, b)