[Rust로 백준 하루 하나] 14-3. 회사에 있는 사람

문제 (7785번)

상근이는 세계적인 소프트웨어 회사 기글에서 일한다. 이 회사의 가장 큰 특징은 자유로운 출퇴근 시간이다. 따라서, 직원들은 반드시 9시부터 6시까지 회사에 있지 않아도 된다.

 

각 직원은 자기가 원할 때 출근할 수 있고, 아무때나 퇴근할 수 있다.

 

상근이는 모든 사람의 출입카드 시스템의 로그를 가지고 있다. 이 로그는 어떤 사람이 회사에 들어왔는지, 나갔는지가 기록되어져 있다. 로그가 주어졌을 때, 현재 회사에 있는 모든 사람을 구하는 프로그램을 작성하시오.

 

입력

첫째 줄에 로그에 기록된 출입 기록의 수 n이 주어진다. (2 ≤ n ≤ 106) 다음 n개의 줄에는 출입 기록이 순서대로 주어지며, 각 사람의 이름이 주어지고 "enter"나 "leave"가 주어진다. "enter"인 경우는 출근, "leave"인 경우는 퇴근이다.

 

회사에는 동명이인이 없으며, 대소문자가 다른 경우에는 다른 이름이다. 사람들의 이름은 알파벳 대소문자로 구성된 5글자 이하의 문자열이다.

 

출력

현재 회사에 있는 사람의 이름을 사전 순의 역순으로 한 줄에 한 명씩 출력한다.

---

풀이

코드

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

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 line = String::new();
    let mut employees = HashSet::new();
    for _ in 0..n {
        line.clear();
        reader.read_line(&mut line).unwrap();
        let mut line_iter = line
            .trim()
            .split_whitespace();
        let name = line_iter.next().unwrap();
        let attendance = line_iter.next().unwrap();

        if attendance == "enter" {
            employees.insert(name.to_string());
        } else {
            employees.remove(name);
        }
    }

    let mut sorted_employees: Vec<String> = employees.into_iter().collect();
    sorted_employees.sort_by(|a, b| b.cmp(a));
    for employee in sorted_employees  {
        writeln!(writer, "{}", employee).unwrap();
    }
}

해설

처음에 HashSet과 HashMap에 정렬 메서드가 없다는 것에서 막혀서 다음과 같이 벡터를 사용해서 풀었는데 시간초과에 걸리게 되었다. 숫자가 클때는 항상 position에서 많은 시간을 잡아먹는 것 같습니다.

 

이런 경우 탐색, 삽입, 삭제를 HashSet과 HashMap에서 진행해준 후 결과 출력시에 이터레이터를 통해 벡터로 만들어서 정렬해서 출력하면 됩니다.

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 line = String::new();
    let mut log: Vec<String> = Vec::new();
    for _ in 0..n {
        line.clear();
        reader.read_line(&mut line).unwrap();
        let mut line_iter = line
            .trim()
            .split_whitespace();
        let name = line_iter.next().unwrap();
        let attendance = line_iter.next().unwrap();

        if attendance == "enter" {
            log.push(name.to_string());
        } else {
            let index = log.iter().position(|x| *x == name).unwrap();
            log.remove(index);
        }
    }
    log.sort_unstable();
    log.reverse();
    for person in log {
        writeln!(writer, "{}", person).unwrap();
    }
}

---

추가 학습

• 특이사항 없음