마구간 문제

밑의 코드와 같이 작성하여 실행시켰는데, 2번 예제에 대해서만 오류가 뜹니다.

정답은 43이라고 되어 있는데, 제 답은 37로 나옵니다.

강사님의 강의와 제 코드를 비교해봐도 어느 부분이 정확히 틀렸는지 잘 모르겠습니다.

검토해주시고 알려주시면 정말 감사하겠습니다.

import sys

sys.stdin = open(r"C:\Users\user\Desktop\Documents\파이썬 알고리즘 문제풀이(코딩테스트 대비)\섹션 4\4. 마구간 정하기\in2.txt", "r")
n ,target_count = list(map(int, input().split()))
input_list1 = []
for i in range(n):

    input_list1.append(int(input()))


def decide_horse_house():

    input_list = sorted(input_list1)

    left = 1
    right = max(input_list) - 1
    distance_between_closet_horses = (left+right)//2
    result = 0
    
    while True:
        
        count = 1
        start = input_list[0]
        for i in range(start+1, len(input_list)):
            
            if (input_list[i] - start) >= distance_between_closet_horses:
                
                count = count + 1
                start = input_list[i]
                
        if count >= target_count:
            
            result = distance_between_closet_horses
            left = distance_between_closet_horses + 1
            distance_between_closet_horses = (left+right)//2
            
        elif count < target_count:
            
            right = distance_between_closet_horses - 1
            distance_between_closet_horses = (left+right)//2
        
        if left > right:
                
            print(result)
            break
        
decide_horse_house() 

인프런 커뮤니티 질문&답변