[백준] 10799번 쇠막대기

www.acmicpc.net/problem/10799

10799번: 쇠막대기

#include <iostream>
#include <string>
using namespace std;

int main() {
	string s;
	int cnt = 0; // 현 시점에 있는 쇠막대기
	int stickNum = 0; // 현재까지 잘린 막대기 조각 개수
	cin >> s;
	for (int i = 0; i < s.size(); i++) {
		if (s[i] == '(') {
			if (i < s.size() - 1 && s[i + 1] == ')') { // 레이저일때 -> 상황 A
				stickNum += cnt;
				i++; // 레이저이므로 다음 s[i]는 ')'이므로 다음 문자는 뛰어넘음
			}
			else { // 여는 괄호일 때 -> 상황 B
				cnt++;
			}
		}
		else { // 닫는 괄호일 때 -> 상황 C
			cnt--;
			stickNum++;
		}
	}
	cout << stickNum;
	return 0;
}

일단 문자열로 받은 다음 문자열의 길이만큼 for문을 돌면서 순차적으로 그 시점에 있는 쇠막대기의 개수와 그 시점에서 현재까지 잘린 막대기 조각의 개수를 세었다.

현 시점에 있는 쇠막대기의 개수: cnt

현재까지 잘린 막대기 조각의 개수: stickNum

()(((()())(())()))(())

입력이 위와 같이 들어왔을 때

위와 같이 동작한다. 따라서 답은 17이 된다.

상황 A는 if문 안에서의 if문 (레이저일 때)

상황 B는 if문 안에서의 else문 (여는 괄호일 때)

상황 C는 else문 (닫는 괄호일 때)

 

코드에 조금 더 충실하게 표를 그리자면

위와 같다. 레이저일 경우 i++로 건너뛰는 것을 표현해주었다.