AtCoderABC142 Dで一つだけWAになる原因

Question

前提・実現したいこと

問題はこちらです
ソースコードを提出した際に、一つだけ（testcase 24）がWA判定になります。自分では原因が特定できませんでした。

import math

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

def judge_prime_number(target):
    dest = int(math.sqrt(target))
    for i in range(2, dest +1):
        if target % i == 0:
            return False
    return True

def return_divisor(target):
    divisor_set = set()
    dest = int(math.sqrt(target))
    for i in range(2, dest + 1):
        if target % i == 0:
            if judge_prime_number(i):
                divisor_set.add(i)
            if judge_prime_number(target // i):
                divisor_set.add(target // i)
    if target != 0:
        divisor_set.add(1)
    return divisor_set
            

a_divisor_set = return_divisor(a)
b_divisor_set = return_divisor(b)
common_divisor = a_divisor_set & b_divisor_set
print(len(common_divisor))

していることは、return_divisor関数で素数である約数の集合を生成します。素数の判定はjudge_prime_number関数で行っています。

半日ほど考えましたが、原因がつかめませんでした。お気づきになられた点がありましたら、よろしくおねがいいたします。

Answer 1

def return_divisor(target):

がtarget自身が素数であるときに、target自身を含まないようになっています。

return_divisor(3) #=> {1}

そのせいでどちらかいっぽうが素数で、その素数が公約数になるようなテストケースを落としてるんでしょう

Comments

[sub_machida]

解決しました！どうもありがとうございます