본문 바로가기

알고리즘 · 코딩

[프로그래머스] 쇠막대기

문제 링크

https://programmers.co.kr/learn/courses/30/lessons/42585

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr


C++ 풀이

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

int solution(string arrangement) {
	int answer = 0; //쇠막대기 수 + 커팅 횟수
	int currentStickN = 0; //레이저가 자르게 될 쇠막대기 개수
	for (int i = 0; i < arrangement.length() - 1; i++) {
		if (arrangement[i] == '(') {
			if (arrangement[i + 1] == ')') { //레이저인 경우
				answer += currentStickN;
			}
			else { //쇠막대기 시작 위치인 경우
				answer++;
				currentStickN++;
			}
		}
		else if ((arrangement[i] == ')') && (arrangement[i - 1] == ')')) { //쇠막대기 끝나는 위치인 경우
			currentStickN--;
		}
	}
	return answer;
}

왼쪽부터 차례대로 문자열을 읽으며, 레이저가 나올 때마다 현재 놓여진 막대기 개수를 더하도록 구현하였다.

 

 

 

반응형