백준 C++ 10799번) 쇠막대기

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

10799번: 쇠막대기

 

Stack STL를 활용하여 풀었습니다. 길이를 활용하여 '(' 뒤에 바로 ')'가 나오면 stack의 사이즈를 체크하여 잘려진 만큼 더하였고, 그 이외의 경우에는 그냥 pop을 하여 막대기에서 제외하였습니다.

 

의사코드

if( c  ==  '(' ){

    푸쉬하여 넣고 갯수를 하나 늘린다. 그리고 (를 넣엇다는 것을 체크해준다.

}

else if ( c == ')' ){

    if( '('뒤에 바로 오는지 체크 ){

        레이저로 자르는 것이기 때문에 스택에서 하나 뺀 갯수를 total에 더해준다.

    }else{

        그냥 스택에서 하나 뺀다

    }

    ( 를 체크 해제한다.

}

 

#include<iostream>
#include<string>
#include<stack>

using namespace std;

int main() {
	ios::sync_with_stdio(0);
	cin.tie(nullptr);
	string s;
	stack<char> bal;
	int total = 0;
	int len = 0;
	getline(cin, s);
	for (auto c : s) {
		if (c == '(') {
			bal.push(c);
			total += 1;
			len = 0;
		}
		else if (c == ')') {
			if (len == 0) {
				bal.pop();
				total += bal.size();
				total -= 1;
			}
			else {
				bal.pop();
			}
			len = 1;
		}
	}
	cout << total << "\n";
}