티스토리 뷰
반응형
문제 링크
문제
정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.
명령은 총 다섯 가지이다.
- push X: 정수 X를 스택에 넣는 연산이다.
- pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
- size: 스택에 들어있는 정수의 개수를 출력한다.
- empty: 스택이 비어있으면 1, 아니면 0을 출력한다.
- top: 스택의 가장 위에 있는 정수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
입력
첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.
출력
출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.
문제 풀이
5가지 명령에 대한 조건문을 작성하고 해당 명령마다 적합한 실행을 하면 된다.
소스 코드
import Foundation
let N = Int(readLine()!)
var stack: Array<Int> = [Int]()
for _ in 0..<N! {
let input = readLine()!.split(separator: " ").map(){String($0)}
if input[0] == "push"{
stack.append(Int(input[1])!)
} else if input[0] == "top"{
if !stack.isEmpty {
print(stack[stack.count - 1])
} else {
print(-1)
}
} else if input[0] == "size"{
print(stack.count)
} else if input[0] == "pop"{
if !stack.isEmpty {
print(stack.remove(at: stack.count - 1))
} else {
print(-1)
}
} else if input[0] == "empty"{
if stack.isEmpty {
print(0)
} else {
print(1)
}
}
}
반응형
'Algorithm > BaekJoon' 카테고리의 다른 글
[백준] 2630번 색종이 만들기 [Swift] (0) | 2020.07.31 |
---|---|
[백준] 1931번 회의실배정 with Swift (1) | 2020.07.29 |
[백준] 10816번 숫자 카드 2 [Swift] (0) | 2020.07.29 |
[백준] 1260번 DFS와 BFS [Swift] (0) | 2020.07.28 |
[백준] 1920번 수 찾기[Swift] (0) | 2020.07.28 |
댓글
공지사항
최근에 올라온 글
최근에 달린 댓글
- Total
- Today
- Yesterday
링크
TAG
- 동시성
- 테이블뷰
- OS
- Apple
- dfs
- 자료구조
- pattern
- DP
- document
- System
- Swift
- 아이폰
- design
- BFS
- operating
- OSTEP
- 앱개발
- Publisher
- 백준
- IOS
- 스위프트
- 코테
- 문법
- operator
- 프로그래밍
- Xcode
- mac
- 코딩테스트
- Combine
- 알고리즘
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
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 |
글 보관함