프로그래머스_스택/큐_쇠막대기 (JAVA)

문제 설명 여러 개의 쇠막대기를 레이저로 절단하려고 합니다. 효율적인 작업을 위해서 쇠막대기를 아래에서 위로 겹쳐 놓고, 레이저를 위에서 수직으로 발사하여 쇠막대기들을 자릅니다. 쇠막대기와 레이저의 배치는 다음 조건을 만족합니다. (a) 레이저는 여는 괄호와 닫는 괄호의 인접한 쌍 '()'으로 표현합니다. 또한 모든 '()'는 반드시 레이저를 표현합니다. (b) 쇠막대기의 왼쪽 끝은 여는 괄호 '('로, 오른쪽 끝은 닫힌 괄호 ')'로 표현됩니다.   아래 그림은 위 조건을 만족하는 예를 보여줍니다. 수평으로 그려진 굵은 실선은 쇠막대기이고, 점은 레이저의 위치, 수직으로 그려진 점선 화살표는 레이저의 발사 방향입니다.       이러한 레이저와 쇠막대기의 배치는 다음과 같이 괄호를 이용하여 왼쪽부터 순서대로 표현할 수 있습니다.   위 예의 괄호 표현은 그림 위에 주어져 있습니다. 쇠막대기는 레이저에 의해 몇 개의 조각으로 잘리는데, 위 예에서 가장 위에 있는 두 개의 쇠막대기는 각각 3개와 2개의 조각으로 잘리고, 이와 같은 방식으로 주어진 쇠막대기들은 총 17개의 조각으로 잘립니다. 쇠막대기와 레이저의 배치를 표현한 문자열 arrangement가 매개변수로 주어질 때, 잘린 쇠막대기 조각의 총 개수를 return 하도록 solution 함수를 작성해주세요.   제한사항 arrangement의 길이는 최대 100,000입니다. arrangement의 여는 괄호와 닫는 괄호는 항상 쌍을 이룹니다. 입출력 예   입출력 예 설명 문제에 나온 예와 같습니다.

 

 

다항식 연산 로직을 작성해보신분들은 아시겠지만 괄호 검증에 대한 문제는 대부분 스택을 이용해 풀게 됩니다. 열린괄호 갯수대로 닫힌 괄호가 나와야하고, 닫힌 괄호가 나올 때마다 스택에 쌓여있던 열린 괄호를 하나씩 제거해주다보면 언젠가 0이 되는 구조이기 때문입니다.

 

물론 이 문제는 괄호에 대한 검증이 아니기 때문에 단순히 스택 구조만 이용해 문제를 풀면 됩니다.

 

 

---

 

 

요점은 레이져가 발사되는 시점에 쇠막대기가 몇 개가 놓여있는가 하는 것입니다. 레이져가 발사되면 현재 놓여진 막대기 갯수만큼 막대기가 늘어나고, 막대기가 끝나면 레이져가 있건 없건 1개가 늘어납니다. 이 부분만 잘 이용하면 어렵지 않게 로직 작성이 가능합니다.

 

import java.util.Stack;

class Solution {

	public int solution(String arrangement) {

		int answer = 0;
		Stack<Integer> stack = new Stack<>();
		int len = arrangement.length();

		for (int i = 0; i < len; i++) {

			if (arrangement.charAt(i) == '(') {
				stack.push(i);

			} else {

				if (stack.peek() + 1 == i) {
					stack.pop();
					answer += stack.size();

				} else {
					stack.pop();
					answer += 1;
				}
			}
		}

		return answer;
	}
}

 

 

---

 

 

직관적으로 아래와 같은 코드로도 간단히 짤 수 있습니다.

 

class Solution {

	public int solution(String arrangement) {

		int barCount = 0;
		int answer = 0;
		int size = arrangement.length();

		for (int i = 0; i < size; i++) {

			if (arrangement.charAt(i) == '(') {

				if (i < size - 1 && arrangement.charAt(i + 1) == ')') {
					answer += barCount;

				} else {
					answer++;
					barCount++;
				}

			} else {

				if (i > 0 && arrangement.charAt(i - 1) != '(') {
					barCount--;
				}
			}
		}

		return answer;
	}
}