[테코테코1.5 2-05] 막대기

💡테코테코 시즌 1.5 2회 모임 off-site 문제 풀이입니다. (2025.01.19)

 

문제

 

출처: 백준 - 막대기

 

내용

 

아래 그림처럼 높이만 다르고 (같은 높이의 막대기가 있을 수 있음) 모양이 같은 막대기를 일렬로 세운 후, 왼쪽부터 차례로 번호를 붙인다. 각 막대기의 높이는 그림에서 보인 것처럼 순서대로 6, 9, 7, 6, 4, 6 이다. 일렬로 세워진 막대기를 오른쪽에서 보면 보이는 막대기가 있고 보이지 않는 막대기가 있다. 즉, 지금 보이는 막대기보다 뒤에 있고 높이가 높은 것이 보이게 된다. 예를 들어, 그림과 같은 경우엔 3개(6번, 3번, 2번)의 막대기가 보인다.

N개의 막대기에 대한 높이 정보가 주어질 때, 오른쪽에서 보아서 몇 개가 보이는지를 알아내는 프로그램을 작성하려고 한다.

 

 

기록하기

💡 어디까지 생각해봤는지 단계적으로 기록해봅니다.

어떤 알고리즘을 적용하려 했나요?

스택을 사용한다.

적용 근거는 무엇인가요?

가장 가까운, 혹은 가장 최근의 값을 비교할 때 스택이 용이하다.

풀이

풀이 시간

시작 시각	종료 시각	총 소요 시간
17:45	17:56	11분

 

문제 분석

 

제약 사항 파악 & 입력값 분석

💡 입력값을 분석하면 문제에서 요구하는 알고리즘의 시간 복잡도를 간접적으로 파악할 수 있습니다.

 

첫 번째 줄에는 막대기의 개수를 나타내는 정수 N (2 ≤ N ≤ 100,000)이 주어지고 이어지는 N줄 각각에는 막대기의 높이를 나타내는 정수 h(1 ≤ h ≤ 100,000)가 주어진다.

6
6
9
7
6
4
6

 

오른쪽에서 N개의 막대기를 보았을 때, 보이는 막대기의 개수를 출력한다.

3

 

 

의사 코드 작성

💡 의사 코드는 동작 중심으로 작성하는 것이 중요합니다.

💡 의사 코드는 문제 해결 순서로 작성합니다.

💡 의사 코드를 충분히 테스트해봅니다.

soltion() {
    // 1. 막대기 개수 int N 받기
    // 2. 스택 생성
    for (int i = 0; i < N; i++) {
        // 3. 막대기 높이 int h 받기
        while (스택에 값 있고 h > top 일때) {
            stack.pop();
        }

        if (스택이 비어있거나 h < top 이면) {
            stack.push(h);
        }
    }
    System.out.println(stack.size());
}

 

구현

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Stack;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(bufferedReader.readLine());
        Stack<Integer> stack = new Stack<>();
        for (int i = 0; i < N; i++) {
            int h  = Integer.parseInt(bufferedReader.readLine());
            while (!stack.isEmpty() && h > stack.peek()) {
                stack.pop();
            }
            if (stack.isEmpty() || h < stack.peek()) {
                stack.push(h);
            }
        }
        System.out.println(stack.size());
    }
}

 

요약하기

• 이게 바로 모노토닉 스택..!