34-H0ngJu

[H0ngJu]

🔗 문제 링크

개똥벌레

✔️ 소요된 시간

1시간??

✨ 수도 코드

이 문제는 킥은
(2 ≤ N ≤ 200,000, 2 ≤ H ≤ 500,000)

이 조건입니다.

사실 문제 자체는 완탐으로도 풀 수는 있지만,
200,000 * 500,000 으로 풀어버리면 시간초과가 발생합니다.

그래서 시간을 줄이기 위한 접근으로
가장 먼저 이분탐색을 생각했습니다.

mid를 잡고 로직을 구현하는 과정에서 여기서 구해지는 mid가 뭐가 되어야 하는지 잘 모르겠더라고요

이분탐색을 제외하고 시간을 줄일 수 있는 방법은 뭐가있지,,
하고 알고리즘 분류를 보니까 누적합 이 있더라고요

근데 why? 누적합으로 시간을 줄일 수 있는가?

# info[odd] < fly and H - info[even] + 1 > fly 인 구간 중 최소를 찾으면 됨
# for fly in range(1,H+1):
#     broke = 0
#     for i in range(N):
#         if i%2 == 0: # 짝수인 경우
#             if H-info[i] + 1 > fly:
#                 continue
#             else:
#                 broke += 1
#         else:
#             if info[i] < fly:
#                 continue
#             else:
#                 broke += 1
#     if broke == min_broke:
#         broke_cnt += 1
#     elif broke < min_broke:
#         min_broke = broke
#         broke_cnt = 1

여기서 모든 석순,종유석을 다 살펴보기 때문에,
위에서 말한대로 200,000 * 500,000가 되어(이중 for문) 시간초과가 발생하지만,

길이가 N개 이상인 석순과 종유석을 계산해놓으면,
2*100,000 + 500,000가 되어 시간초과를 막을 수 있습니다.

시간초과 -> 이중for문이 대부분 문제..

그래서 누적합으로 접근하면,

석순과 종유석을 따로 분리해서
길이가 n 이상인 것들의 합을 저장해줍니다.

for i in range(H-1, 0, -1):
    down[i] += down[i+1]
    up[i] += up[i+1]

개똥벌레가 h-1 ~ h 사이의 구간을 날 때,
파괴해야하는 석순과 종유석은
석순[h] (길이가 h이상), 종유석[동굴길이-h]이므로

for i in range(1, H+1):
    broke = up[i] + down[H-i+1]
    if broke < min_broke:
        min_broke = broke
        broke_cnt = 1
    elif broke == min_broke:
        broke_cnt += 1

요로코롬 최솟값을 갱신해주면 됩니다.

누적합으로 문제를 풀고, 다시 이분탐색으로 구글링해서 보니까
상한선, 하한선 개념으로 mid를 구해서 둘을 더하는 식으로 구현할 수 있네용

📚 새롭게 알게된 내용

[tgyuuAn]

이 문제는 킥은

(2 ≤ N ≤ 200,000, 2 ≤ H ≤ 500,000)

이 조건입니다.

ㅋㅋㅋㅋㅋㅋㅋㅋㅋㅋㅋㅋ

저도 누적합 바로 떠오르고 풀었는데,,

from collections import defaultdict
import sys

def input(): return sys.stdin.readline().rstrip()

N, H = map(int, input().split())

floors = []
bottoms = []
for idx in range(N):
    height = int(input())
    height *= -1
        
    if idx%2==0: floors.append(height)
        
    else: bottoms.append(height)

floors.sort()
bottoms.sort()

f_prefix_sum = defaultdict(int)
p_f_h = 0
for f_h in floors:
    if p_f_h == 0 or p_f_h == f_h:
        f_prefix_sum[f_h] += 1
    else:
        f_prefix_sum[f_h] = f_prefix_sum[p_f_h] + 1
        
    p_f_h = f_h

b_prefix_sum = defaultdict(int)
p_b_h = 0
for b_h in bottoms:
    if p_b_h == 0 or p_b_h == b_h:
        b_prefix_sum[b_h] += 1
    else:
        b_prefix_sum[b_h] = b_prefix_sum[p_b_h] + 1
    
    p_b_h = b_h

b_temp = []
for b_h in range(H, 0, -1):
    if -1*b_h in b_prefix_sum: b_temp.append(b_prefix_sum[-1*b_h])
    else: 
        if b_temp: b_temp.append(b_temp[-1])
        else: b_temp.append(0)
b_temp.reverse()        

f_temp = []
for f_h in range(H, 0, -1):
    if -1*f_h in f_prefix_sum: f_temp.append(f_prefix_sum[-1*f_h])
    else: 
        if f_temp: f_temp.append(f_temp[-1])
        else: f_temp.append(0)

answer = 0
min_count = int(1e9)
for idx, (b, f) in enumerate(zip(b_temp, f_temp)):
    temp = b+f
    if min_count > temp:
        answer = 1
        min_count = temp
    elif min_count == temp:
        answer += 1

print(f"{min_count} {answer}")

제 코드보다 홍쥬르 코드가 더 간결하고 이쁘네요 ,,, ㅠㅠㅠㅠ