[백준/C++] 10799번 쇠막대기

문제: www.acmicpc.net/problem/10799

10799번: 쇠막대기

 

 

문제풀이

 

왼쪽부터 레이저를 만나면 현재까지의 쇠막대기 수(스택의 사이즈 -1 )를 결과값에 더해주고 쇠막대기의 끝에 도달하면 결과값에 1을 더해주었다.

 

( ( -> s.push(1)

( ) ->  result += s.size() - 1

) ) -> result++

 

 

코드

#include <iostream>
#include <stack>

using namespace std;

int main(){
    ios::sync_with_stdio(false); cin.tie(nullptr);

    string str;
    cin>>str;

    int result = 0;
    stack<int> s;
    for(int i=0; i<str.length(); i++){
        if(str[i] == '('){
            s.push(1);
        }
        else{
            if(str[i-1] == '('){ //레이저로 현재까지의 쇠막대기를 자름 
                result += s.size()-1;
            }
            else{ //막대의 끝이므로 조각 한 개 플러스
                result++;
            }
            s.pop();
        }
    }

    cout<<result;

    return 0;
}

 

 

결과