[백준] 2841번 외계인의 기타 연주

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

2841번: 외계인의 기타 연주

stack을 이차원으로 선언하여

줄 번호에 따른 프렛의 번호를 push해주면 된다.

 

같은 줄일 때

 

1. stack의 top > 현재 프렛

stack의 top < 현재 프렛 될때까지 pop (손가락 떼는 것 +1)

현재 프렛 push (손가락 누르는 것 +1)

 

2. stack의 top < 현재 프렛

현재 프렛만 push 해준다. (손가락 누르는 것 +1)

 

3. stack의 top = 현재 프렛

이 경우는 카운트해줄 필요 없이 넘어간다.
     

이 조건들을 구현해주면 끝

#include <iostream>
#include <stack>
using namespace std;
stack <int> melody[7];

int main() {
	int N, P;
	cin >> N >> P;

	int n, p;
	long long res = 0;
	for (int i = 0; i < N; i++) {
		cin >> n >> p;
		if (melody[n].empty()) {
			melody[n].push(p);
			res++;
		}
		else {
			while (!melody[n].empty() && melody[n].top() > p) {
				melody[n].pop();
				res++;
			}
			if (melody[n].empty() || melody[n].top() != p) {
				melody[n].push(p);
				res++;
			}
		}
	}

	cout << res;

	return 0;
}