[BOJ][🟡5][백준#03020] 개똥벌레

문제 출처

BAEKJOON Online Judge #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

출력

2 3

예제 2

입력

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

출력

7 2

My Sol

import sys
input = sys.stdin.readline

N, H = map(int, input().split())
ceil = [0]*(H+1)
floor = [0]*(H+1)
for _ in range(N//2):
    n = int(input())
    floor[n] += 1
    n = int(input())
    ceil[n] += 1

min_collapse = cur_collapse = N//2
cur_cnt = 1
for h in range(2, H+1):
    cur_collapse += (ceil[H+1-h] - floor[h-1])
    if cur_collapse > min_collapse: continue
    if cur_collapse < min_collapse:
        min_collapse = cur_collapse
        cur_cnt = 1
    else: cur_cnt +=1

print(min_collapse, cur_cnt)

문제에서 제시하는 문제 풀이는 이분탐색 또는 부분합 알고리즘을 사용하는 문제로 제시되어 있는데, 사실상 알고리즘이 필요 없는 문제였다.

1. 입력을 처리한다. 천장(ceil)과 바닥(floor)을 각각 구분해 입력을 받는다. 입력이 크므로 sys.stdin.readline 으로 입력을 처리한다.
2. 장애물은 0보다 크고 H보다 작으므로, 첫 줄과 끝 줄은 항상 N//2의 장애물을 가진다. 때문에 이 수로 초기화 해준다.
3. 아래부터 위로 올라가는데, 해당 구간에서 새롭게 발견된, 즉 마주치는 종유석은 ceil에서 계산해 현재 충돌에 더해주고, 반대로 석순은 이전 구간까지 크기가 있는 것을 빼주어 충돌에서 벗어났음을 계산한다.
4. 만약 현재 충돌의 수가 최소 충돌보다 크면 무의미하므로 continue처리하고, 작으면 해당 구간의 충돌 수를 기록하며 해당하는 충돌의 수가 1개 구간이라는 의미로 cur_cnt를 1로 처리한다. 만약 같으면 cur_cnt만 하나 올려주면 되겠다.
5. 이를 마치고 min_collapse 와 cur_cnt를 각각 출력한다.

결과

맞았습니다!!