스택/큐 문제는 문제를 풀수록 잘 모르겠는 것이, ‘왜 이 문제가 스택/큐 문제이지?’ 하는 의문이 든다. 스택이나 큐 자료구조를 사용하지 않아도 풀 수 있고, 또 실제로 그렇게 풀고있으니까 말이다. 물론 이전에도 말했듯이 알고리즘 문제를 풀 때는 자료구조를 어떤 것을 사용해야한다는 제한이나 조건이 없기 때문에 자유롭게 문제를 풀 수 있지만 말이다. 본 문제는 푸는데 시간이 엄청 짧게 걸렸고 그래서 기분이 좋았던 문제였다 :)
1. 문제
여러 개의 쇠막대기를 레이저로 절단하려고 합니다.
효율적인 작업을 위해서 쇠막대기를 아래에서 위로 겹쳐 놓고, 레이저를 위에서 수직으로 발사하여 쇠막대기들을 자릅니다.
쇠막대기와 레이저의 배치는 다음 조건을 만족합니다.
1
2
3
4
- 쇠막대기는 자신보다 긴 쇠막대기 위에만 놓일 수 있습니다.
- 쇠막대기를 다른 쇠막대기 위에 놓는 경우 완전히 포함되도록 놓되, 끝점은 겹치지 않도록 놓습니다.
- 각 쇠막대기를 자르는 레이저는 적어도 하나 존재합니다.
- 레이저는 어떤 쇠막대기의 양 끝점과도 겹치지 않습니다.
아래 그림은 위 조건을 만족하는 예를 보여줍니다.
수평으로 그려진 굵은 실선은 쇠막대기이고, 점은 레이저의 위치, 수직으로 그려진 점선 화살표는 레이저의 발사 방향입니다.
이러한 레이저와 쇠막대기의 배치는 다음과 같이 괄호를 이용하여 왼쪽부터 순서대로 표현할 수 있습니다.
1
2
(a) 레이저는 여는 괄호와 닫는 괄호의 인접한 쌍 '()'으로 표현합니다. 또한 모든 '()'는 반드시 레이저를 표현합니다.
(b) 쇠막대기의 왼쪽 끝은 여는 괄호 '('로, 오른쪽 끝은 닫힌 괄호 ')'로 표현됩니다.
위 예의 괄호 표현은 그림 위에 주어져 있습니다.
쇠막대기는 레이저에 의해 몇 개의 조각으로 잘리는데, 위 예에서 가장 위에 있는 두 개의 쇠막대기는 각각 3개와 2개의 조각으로 잘리고, 이와 같은 방식으로 주어진 쇠막대기들은 총 17개의 조각으로 잘립니다.
쇠막대기와 레이저의 배치를 표현한 문자열 arrangement가 매개변수로 주어질 때,
잘린 쇠막대기 조각의 총 개수를 return 하도록 solution 함수를 작성해주세요.
제한조건
- arrangement의 길이는 최대 100,000입니다.
- arrangement의 여는 괄호와 닫는 괄호는 항상 쌍을 이룹니다.
2. 알고리즘! 생각해보자
arrangement를 돌면서, ‘(‘이 나올때와 ‘)’이 나올 때로 경우를 나눈다.- ’(‘이 나오면 그 다음 문자가 ‘)’인지 확인해서 레이저인지 아닌지 확인한다.
- 레이저일 경우 현재 파이프를 절단했다는 의미이므로, 현재 pipe 수를 나타내는
pipe_count만큼answer에 더한다. - 또한 레이저일 경우 다음 문자 ‘)’는 이미 해석했으므로
i를 증가시킨다. - 레이저가 아닐 경우 쇠막대기라는 의미이므로
pipe_count를 증가시킨다. - 문자가 ‘)’일 경우는 쇠막대기가 끝났을 경우이므로
pipe_count를 감소시킨다. - 또한 쇠막대기가 끝났을 경우 마지막 절단하고 남은 마지막 조각이 있으므로
answer를 1만큼 증가시킨다. answer를 반환한다.
3. 해결코드
[C++]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
#include <string>
#include <vector>
using namespace std;
int solution(string arrangement) {
int pipe_count = 0;
int answer = 0;
for (int i = 0; i < arrangement.size(); i++) {
if (arrangement[i] == '(') {
if (arrangement[i + 1] == ')') {
answer += pipe_count;
i++;
} else pipe_count++;
} else if (arrangement[i] == ')') {
answer++;
pipe_count--;
}
}
return answer;
}
[Javascript]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
function solution(arrangement) {
var pipe = 0;
var answer = 0;
for (var i = 0; i < arrangement.length; i++) {
if (arrangement[i] == "(" && arrangement[i + 1] == ")") {
answer += pipe;
i++;
} else if (arrangement[i] == ")") {
answer++;
pipe--;
} else if (arrangement[i] == "(") pipe++;
}
return answer;
}
4. 해결능력UP, 깊이 생각해보기
- 잘 푼것 같아서 넘나 뿌듯…!
- 하지만 스택/큐는 사용하지 않았다..
- 다른 사람이 푼 방법을 봤는데 나와 같은 로직을 사용했지만
stack을 사용한 사람이 있었다
5. 참고해서 문제해결 ٩( ᐛ )و
- 처음으로 없다! 스스로 해결 :)
1
출처: 프로그래머스 코딩 테스트 연습, https://programmers.co.kr/learn/challenges