[BOJ 3020 🥇5] 개똥벌레 (Python)

https://www.acmicpc.net/problem/3020

3020번: 개똥벌레

문제

개똥벌레 한 마리가 장애물(석순과 종유석)로 가득찬 동굴에 들어갔다. 동굴의 길이는 N미터이고, 높이는 H미터이다. (N은 짝수) 첫 번째 장애물은 항상 석순이고, 그 다음에는 종유석과 석순이 번갈아가면서 등장한다.

아래 그림은 길이가 14미터이고 높이가 5미터인 동굴이다. (예제 그림)

이 개똥벌레는 장애물을 피하지 않는다. 자신이 지나갈 구간을 정한 다음 일직선으로 지나가면서 만나는 모든 장애물을 파괴한다.

위의 그림에서 4번째 구간으로 개똥벌레가 날아간다면 파괴해야하는 장애물의 수는 총 여덟개이다. (4번째 구간은 길이가 3인 석순과 길이가 4인 석순의 중간지점을 말한다)

하지만, 첫 번째 구간이나 다섯 번째 구간으로 날아간다면 개똥벌레는 장애물 일곱개만 파괴하면 된다.

동굴의 크기와 높이, 모든 장애물의 크기가 주어진다. 이때, 개똥벌레가 파괴해야하는 장애물의 최솟값과 그러한 구간이 총 몇 개 있는지 구하는 프로그램을 작성하시오.

 

입력

첫째 줄에 N과 H가 주어진다. N은 항상 짝수이다. (2 ≤ N ≤ 200,000, 2 ≤ H ≤ 500,000)

다음 N개 줄에는 장애물의 크기가 순서대로 주어진다. 장애물의 크기는 H보다 작은 양수이다.

 

출력

첫째 줄에 개똥벌레가 파괴해야 하는 장애물의 최솟값과 그러한 구간의 수를 공백으로 구분하여 출력한다.

 

예제 입력 1 

6 7
1
5
3
3
5
1

 

예제 출력 1 

2 3

 

예제 입력 2 

14 5
1
3
4
2
2
4
3
4
3
3
3
2
3
3

 

예제 출력 2 

7 2

 

⭕ CODE

import sys

input = sys.stdin.readline

from bisect import bisect_left

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

bottom = []
top = []

for i in range(n):
    a = int(input())
    if i % 2 == 0:
        bottom.append(a)
    else:
        top.append(a)

bottom.sort()
top.sort()

result = 2000001
cnt = 0
for i in range(1, h + 1):
    # 총 부시는 개수
    count = 0

    # 아래 석순을 부수는 개수
    # bisect_left를 하는 이유는 같은 높이일경우 부수는게 가능하므로
    count += (n // 2 - bisect_left(bottom, i))

    # 위 종유석을 부수는 개수
    # 마찬가지로 같은 높이일 경우 부수는게 가능하므로 bisect_left
    count += (n // 2 - bisect_left(top, h + 1 - i))

    # 장애물 부시는 개수가 result보다 작다면 갱신
    if count < result:
        result = count
        # 그러한 구간의 수 + 1
        cnt = 1

    # 최소값과 같다면
    elif count == result:
        # 구간의 수 + 1
        cnt += 1

# 파괴하는 장애물의 최솟값과 그러한 구간의 수 출력
print(result, cnt)

✏️ Comment

각 높이마다의 장애물을 부수는 개수를 확인하여 더 작으면 갱신하고 그러한 개수를 구하는 방식으로 풀었다.