티스토리 뷰
SMALL
오랜만에 백준 푸니까 다 까먹은 것 같아서 자료구조부터 다시 하는 중이다... 하.하
오늘의 문제
1874번: 스택 수열
1부터 n까지에 수에 대해 차례로 [push, push, push, push, pop, pop, push, push, pop, push, push, pop, pop, pop, pop, pop] 연산을 수행하면 수열 [4, 3, 6, 8, 7, 5, 2, 1]을 얻을 수 있다.
www.acmicpc.net
문제 요약
- 1부터 n까지의 수를 스택에 push 할 때, pop 하는 순서대로 숫자가 출력되는데 입력으로 특정 수열이 들어왔을 경우 그 수열처럼 출력하기 위해서 스택에 숫자를 어떤 순서대로 push 하고 pop 해야 하는지 +, -로 나타내라. 단, push 할 때는 오름차순을 지켜야 한다.
내 풀이
import Foundation
/// 입력 받은 숫자
let n = Int(readLine()!)!
/// 입력 받은 수열
var numbers: [Int] = []
/// 숫자 스택
var stack: [Int] = []
/// 스택에 오름차순으로 들어가게 되는 숫자
var target = 1
/// 출력될 문구
var result = ""
// n개의 수열 입력 받기
for _ in 0..<n {
numbers.append(Int(readLine()!)!)
}
while true {
// 1부터 n까지의 오름차순 추가가 끝났을 때
if target > n {
// 스택의 맨 위에 있는(= 배열의 맨 끝) 숫자부터 확인
for n in stack.reversed() {
// 스택의 맨 위 숫자가 포커스 하고 있는 수열의 숫자와 같을 때
if n == numbers[0] {
// 수열의 포커스 이동
numbers.removeFirst()
// 스택 pop
result += "-\n"
} else { // 다를 때
// 더 이상 스택에 추가될 숫자도 없는데, 스택에 남아있는 숫자들과 수열의 순서가 맞지 않는 것이므로
// 해당 수열을 만드는 것은 불가능함
// 출력 문구를 "NO"로 대체
result = "NO"
// for문 종료
break
}
}
// while문 종료
break
} else {
// 스택이 비어있지 않고, 현재 포커스 하고 있는 수열(인덱스 0번째)의 숫자와 스택의 top의 값이 같을 때
if !stack.isEmpty && numbers[0] == stack.last {
// 스택을 pop한다 -> 출력 문구에 "-" 추가
result += "-\n"
// 수열의 포커스 이동 -> 맨 앞의 수를 지움으로써 그 다음 숫자가 numbers[0]이 된다
numbers.removeFirst()
// 스택의 맨 위(= 배열의 맨 마지막) 숫자를 지움으로써 pop을 구현
stack.removeLast()
} else { // 스택이 비어있거나 수열의 포커스 숫자와 스택의 top이 다를 때
// 스택에 숫자를 push한다 -> 출력 문구에 "+" 추가
result += "+\n"
// 스택에 오름차순으로 숫자 push
stack.append(target)
// 오름차순을 위해 1씩 숫자 증가
target += 1
}
}
}
// 결과 출력
print(result)
else 문은 그냥 스택의 로직대로 금방 짰는데 if 문을 작성할 때 놓친 부분이 있어서 좀 걸렸다.
1. 스택에 오름차순 추가가 끝났을 때, 스택에 남아있는 숫자들과 남은 수열이 맞는지 확인하는 작업이 꼭 필요하다!
2. 스택에 남아있는 숫자들과 남은 수열이 맞으면 개수대로 "-"를 출력하면 되고, 하나라도 맞지 않는다면 그 수열은 만드는 것이 불가능한 거라서 "NO" 출력하고 바로 반복문 빠져나오기
아니 근데 티스토리 코드블럭 왜 이래! 주석 들여쓰기 다 맞게 적었는데 이상하게 나온다!!!
LIST
'Algorithm' 카테고리의 다른 글
백준 5397번: 키로거 - Swift 풀이 (0) | 2024.04.08 |
---|---|
백준 4949번: 균형잡힌 세상 - Swift 풀이 (1) | 2024.03.25 |
백준 1021번: 회전하는 큐 - Swift 풀이 (0) | 2024.03.23 |
백준 3986번: 좋은 단어 - Swift 풀이 (0) | 2024.03.16 |
백준 10845번: 큐 - Swift 풀이 (0) | 2024.03.15 |
공지사항
최근에 올라온 글
최근에 달린 댓글
- Total
- Today
- Yesterday
링크
TAG
- 큐
- Baekjoon
- Programmers
- 투포인터
- 코테
- 다이나믹프로그래밍
- 알고리즘
- 자료구조
- 백트래킹
- 우선순위큐
- Python
- SQL
- MySQL
- Swift로백준풀기
- 코테준비
- 백준
- 힙
- ios
- BFS
- swea
- 스택
- 정렬
- 프로그래머스
- 그리디
- dp
- Swift
- 이분탐색
- 구현
- ios앱개발
- 코딩테스트
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
1 | 2 | 3 | ||||
4 | 5 | 6 | 7 | 8 | 9 | 10 |
11 | 12 | 13 | 14 | 15 | 16 | 17 |
18 | 19 | 20 | 21 | 22 | 23 | 24 |
25 | 26 | 27 | 28 | 29 | 30 | 31 |
글 보관함