글 읽기 - 파이썬 시간초과

siliconjelly 3년 전 0

아래 코드가 시간초과가 뜨는데 어떻게 개선하면 좋을까요?

# element가 some_list에 있나 확인하는 함수
def binary_search(element, some_list):
    start_index = 0
    end_index = len(some_list) - 1

    while start_index <= end_index:
        mid_index = (start_index + end_index) // 2
        if some_list[mid_index] == element:
            return 1
        elif some_list[mid_index] > element:
            end_index = mid_index - 1
        elif some_list[mid_index] < element:
            start_index = mid_index + 1

    return 0

a_len = int(input(''))
a_list = input('').split() # 리스트 생성, 참고로 다 문자열임
b_len = int(input(''))
b_list = input('').split()

for i in range(b_len): 
    print(binary_search(b_list[i], sorted(a_list)))

ehdrmsl2001 3년 전 1

a_list를 매번 정렬할 필요가 있나요?

siliconjelly 3년 전 0

a_list.sort()를 해주고 for문을 돌리니 문제가 해결되었습니다. 답변 감사합니다!

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

문제

문제

출처

ICPC

파이썬 시간초과

siliconjelly 3년 전 0

ehdrmsl2001 3년 전 1

siliconjelly 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말