파이썬 알고리즘-23 (탐색, 그리디) 랜선 자르기

jiffydev·2020년 8월 31일
0

Algorithm

목록 보기
25/134
post-thumbnail

23.랜선자르기(결정알고리즘)
엘리트 학원은 자체적으로 K개의 랜선을 가지고 있다. 그러나 K개의 랜선은 길이가 제각각이
다. 선생님은 랜선을 모두 N개의 같은 길이의 랜선으로 만들고 싶었기 때문에 K개의 랜선을
잘라서 만들어야 한다. 예를 들어 300cm 짜리 랜선에서 140cm 짜리 랜선을 두 개 잘라내면
20cm 은 버려야 한다. (이미 자른 랜선은 붙일 수 없다.)
편의를 위해 랜선을 자를때 손실되는 길이는 없다고 가정하며, 기존의 K개의 랜선으로 N개의
랜선을 만들 수 없는 경우는 없다고 가정하자. 그리고 자를 때는 항상 센티미터 단위로 정수
길이만큼 자른다고 가정하자. N개보다 많이 만드는 것도 N개를 만드는 것에 포함된다. 이때
만들 수 있는 최대 랜선의 길이를 구하는 프로그램을 작성하시오.

▣ 입력설명
첫째 줄에는 엘리트학원이 이미 가지고 있는 랜선의 개수 K, 그리고 필요한 랜선의 개수 N이
입력된다. K는 1이상 10,000이하의 정수이고, N은 1이상 1,000,000이하의 정수이다. 그리고
항상 K ≦ N 이다. 그 후 K줄에 걸쳐 이미 가지고 있는 각 랜선의 길이가 센티미터 단위의
정수로 입력된다.

▣ 출력설명
첫째 줄에 N개를 만들 수 있는 랜선의 최대 길이를 센티미터 단위의 정수로 출력한다.

▣ 입력예제 1
4 11
802
743
457
539

▣ 출력예제 1
200

예제설명) 802cm 랜선에서 4개, 743cm 랜선에서 3개, 457cm 랜선에서 2개, 539cm 랜선에서 2개를 잘라내 모두 11개를 만들 수 있다.

내 코드

k,n=map(int, input().split())

lst=sys.stdin.readlines()
lst=[int(i) for i in lst]

lt=1
rt=max(lst)
longest=0
while lt<=rt:
    mid=(lt+rt)//2
    cnt=0
    for i in lst:
        cnt+=i//mid
    
    if cnt>=n:
        longest=mid
        lt=mid+1
        
    else:
        rt=mid-1

print(longest)

풀이와 시간차이가 별로 안날거같은데 시간초과가 떠서 확인중.. 에디터에서는 문제가 없었는데 채점 시스템상으로는 readlines로 입력받으면 안되는 것 같다.

풀이

def Count(len):
    cnt=0
    for x in Line:
        cnt+=(x//len)
    return cnt

k, n=map(int, input().split())
Line=[]
res=0
largest=0
for i in range(k):
    tmp=int(input())
    Line.append(tmp)
    largest=max(largest, tmp)
lt=1
rt=largest
while lt<=rt:
    mid=(lt+rt)//2
    if Count(mid)>=n:
        res=mid
        lt=mid+1
    else:
        rt=mid-1
print(res)

반성점

  • 이번에도 이진탐색을 쓰는 내용이었는데 시작을 어떻게 잡아야될지 몰라서 힌트를 봐야했다.

배운 것

  • 이진탐색을 사용해야 하는지 아는 방법: 문제의 답이 주어진 숫자들의 범위 안에 있을 때 사용한다. 이 문제를 예로 들면 답은 항상 1~주어진 숫자의 최댓값 사이에 존재하므로 중간값을 찾는 것을 반복하여 가장 큰 중간값을 출력하면 된다.
profile
잘 & 열심히 살고싶은 개발자

0개의 댓글