글 읽기 - 걸리지않는 예외를 못찾겠어서 질문드립니다.

mraz3068 4년 전 0

시간초과라고 뜰줄예상했는데 틀렸습니다 라고 뜨길래 예외를 다 못잡는코드라는 것인데, 이것저것 예제를 넣어봐도 예외를 못찾겠어서 질문드립니다

알려주시면 감사하겠습니다.

import sys

n = int(sys.stdin.readline().rstrip())
f = []

def is_prime(n: int) -> bool:
    if n < 2:
        return False
    if n in (2, 3):
        return True
    if n % 2 is 0 or n % 3 is 0:
        return False
    if n < 9:
        return True
    k, l = 5, n**0.5
    while k <= l:
        if n % k is 0 or n % (k+2) is 0:
            return False
        k += 6
    return True


def ovl(f):
    for i in f:
        if f.count(i) > 1 and i != 1:
            return 1
    return 0

def happy(d_list, f):
    while 1:
        temp_sum = 0
        if '1' in f:
            return 1

        elif ovl(f) == 1:
            return 0

        else:
            for i in d_list:
                temp_sum += int(i) ** 2
            f.append(str(temp_sum))
            d_list = list(str(temp_sum))

for i in range(0, n):
    l, m = map(str, input().split())
    m_list = list(m)
    if is_prime(int(m)):
        if happy(m_list, f):
            print('{0} {1} YES'.format(l, m))
        else:
            print('{0} {1} NO'.format(l, m))
    else:
        print('{0} {1} NO'.format(l, m))

sait2000 4년 전 1

51번 52번 줄이 똑같은 걸 2번 출력하는 것 같은데요

mraz3068 4년 전 0

아 감사합니다.근데 수정을 해도 여전히 틀렸습니다 라고 출력이 되가지고요 ㅜㅜ

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

걸리지않는 예외를 못찾겠어서 질문드립니다.

mraz3068 4년 전 0

sait2000 4년 전 1

mraz3068 4년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말