| 시간 제한 | 메모리 제한 | 제출 | 정답 | 맞힌 사람 | 정답 비율 |
|---|---|---|---|---|---|
| 2 초 | 128 MB | 89633 | 33262 | 22892 | 37.501% |
문제
도현이의 집 N개가 수직선 위에 있다. 각각의 집의 좌표는 x1, …, xN이고, 집 여러개가 같은 좌표를 가지는 일은 없다.
도현이는 언제 어디서나 와이파이를 즐기기 위해서 집에 공유기 C개를 설치하려고 한다. 최대한 많은 곳에서 와이파이를 사용하려고 하기 때문에, 한 집에는 공유기를 하나만 설치할 수 있고, 가장 인접한 두 공유기 사이의 거리를 가능한 크게 하여 설치하려고 한다.
C개의 공유기를 N개의 집에 적당히 설치해서, 가장 인접한 두 공유기 사이의 거리를 최대로 하는 프로그램을 작성하시오.
입력
첫째 줄에 집의 개수 N (2 ≤ N ≤ 200,000)과 공유기의 개수 C (2 ≤ C ≤ N)이 하나 이상의 빈 칸을 사이에 두고 주어진다. 둘째 줄부터 N개의 줄에는 집의 좌표를 나타내는 xi (0 ≤ xi ≤ 1,000,000,000)가 한 줄에 하나씩 주어진다.
출력
첫째 줄에 가장 인접한 두 공유기 사이의 최대 거리를 출력한다.
풀이
문제 이해
-
가장 인접한 두 공유기 사이의 거리를 가능한 크게 하여 설치 → 거리2, 거리8 (X) / 거리 5 거리 5 (O)
-
구하는 것 = 가장 인접한 두 공유기 사이의 거리의 최대값 → 이걸 이분 탐색
- 이 “최대값”이 의미하는 은 → upper-bound 참고
- upper bound인 이유
- 초과하는 첫 번째 값 == 설치가 불가능해지는 값
- upper bound -1을 하면 설치가 불가능해지기 바로 전 값이 도출됨
실패
def installRouter(dist):
cnt = C-1
i = 0
while cnt > 0:
for j in range(i, N):
if (homes[j] - homes[i]) >= dist:
cnt -= 1
i = j
break
else:
return -1
if homes[i] < homes[-1]: # homes[i]: 마지막에 설치한 집의 위치 < homes[-1]: 가장 먼 집의 위치
return 1
return 0
N,C = map(int, input().split(" "))
homes = []
for _ in range(N):
homes.append(int(input()))
homes.sort()
dl = 1
dr = (homes[-1] - homes[0] + 1) // (C-1) # 최대 거리 / 설치해야하는 공유기 개수
sameList = [0]*dr
while dl < dr:
dm = (dr + dl) // 2
result = installRouter(dm)
if result >= 0:
dl = dm + 1
else:
dr = dm
print(dr)- 워낙에 문제가 많아서 뭐부터 지적해야할지 모르겠음.
성공
N, C = map(int, input().split())
homes = [int(input()) for _ in range(N)]
homes.sort()
dl = 1
dr = (homes[-1] - homes[0]) // (C-1) +1
while dl < dr:
cnt = C - 1
dm = (dr + dl) // 2
# 설치
i = 0
for j in range(i, N):
d = homes[j] - homes[i]
if d >= dm:
cnt -= 1
i = j
if cnt <= 0: # 설치가 많이 되었으면
dl = dm + 1 # 거리를 더 늘려야함
else: # 설치가 적게 되었으면
dr = dm # 거리를 줄여야함.
print(dr-1)- 수정 내용:
- 설치로직 간소화. 처음에는 하나 설치하고 for문 break하고, 다시 for문 들어와서 하나설치 하고의 반복 궅이 그럴 필요 없이 for문 하나에서 기준 집만 바꾸면서 탐색 지속
- dr 초기값을 늘려주기
- 종료조건이 dl >= dr임. 즉. 같아지면 끝나는데. dr은 초기값에서 멈춰있고 dl이 dr과 같은 값으로 업데이트되면, dr(즉 마지막)을 탐색할 수 없음.
- 단 범위를 넘어가는 값이 도출될 수 있으니 결과값으로 다시 리스트 요소에 접근하려면 주의해야함. #Computer #PS