[Rust로 백준 하루 하나] 16-1. 스택 2

문제 (28278번)

정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.

 

명령은 총 다섯 가지이다.

1. 1 X: 정수 X를 스택에 넣는다. (1 ≤ X ≤ 100,000)
2. 2: 스택에 정수가 있다면 맨 위의 정수를 빼고 출력한다. 없다면 -1을 대신 출력한다.
3. 3: 스택에 들어있는 정수의 개수를 출력한다.
4. 4: 스택이 비어있으면 1, 아니면 0을 출력한다.
5. 5: 스택에 정수가 있다면 맨 위의 정수를 출력한다. 없다면 -1을 대신 출력한다.

 

입력

첫째 줄에 명령의 수 N이 주어진다. (1 ≤ N ≤ 1,000,000)

 

둘째 줄부터 N개 줄에 명령이 하나씩 주어진다.

 

출력을 요구하는 명령은 하나 이상 주어진다.

 

출력

출력을 요구하는 명령이 주어질 때마다 명령의 결과를 한 줄에 하나씩 출력한다.

---

풀이

코드

use std::io::{self, BufReader, BufRead, BufWriter, Write};

fn main() {
    let mut reader = BufReader::new(io::stdin().lock());
    let mut writer = BufWriter::new(io::stdout().lock());

    let mut n = String::new();
    reader.read_line(&mut n).unwrap();
    let n: usize = n.trim().parse().unwrap();

    let mut stack: Vec<i32> = Vec::new();
    let mut input = String::new();
    for _ in 0..n {
        reader.read_line(&mut input).unwrap();
        let mut input_iter = input
            .trim()
            .split_whitespace()
            .map(|i| i.parse::<i32>().unwrap());
        let inst = input_iter.next().unwrap();

        match inst {
            1 => {
                let x: i32 = input_iter.next().unwrap();
                stack.push(x);
            },
            2 => {
                if stack.len() == 0 {
                    writeln!(writer, "-1").unwrap();
                } else {
                    let output = stack.pop().unwrap();
                    writeln!(writer, "{output}").unwrap();
                }
            },
            3 => {
                writeln!(writer, "{}", stack.len()).unwrap();
            },
            4 => {
                if stack.len() == 0 { writeln!(writer, "1").unwrap(); }
                else { writeln!(writer, "0").unwrap(); }
            },
            5 => {
                if stack.len() == 0 {
                    writeln!(writer, "-1").unwrap();
                } else {
                    writeln!(writer, "{}", stack[stack.len()-1]).unwrap();
                }
            },
            _ => (),
        }
        input.clear();
    }
}

해설

match 구문에서 다음과 같이 러스트에게 명시적으로 앞의 갈래에 매칭되지 않은 어떠한 값도 사용하지 않을 것이며, 어떠한 코드도 실행하지 않기를 원한다고 알려줄 수 있습니다.

_ => ()

---

추가 학습

• 특이사항 없음