Submission #18719574

---

Source Code Expand

fn main() {
    let (r, w) = (std::io::stdin(), std::io::stdout());
    let mut sc = IO::new(r.lock(), w.lock());

    let mut stack = vec![];
    let mut size = 0;
    let q: usize = sc.read();
    let l: usize = sc.read();
    for _ in 0..q {
        let s: String = sc.read();
        match s.as_str() {
            "Push" => {
                let n: usize = sc.read();
                let m: i64 = sc.read();
                stack.push((n, m));
                size += n;
                if size > l {
                    println!("FULL");
                    return;
                }
            }
            "Pop" => {
                let mut pop: usize = sc.read();
                if size < pop {
                    println!("EMPTY");
                    return;
                }
                size -= pop;
                while let Some((count, m)) = stack.pop() {
                    if pop < count {
                        let add = count - pop;
                        stack.push((add, m));
                        pop = 0;
                    } else {
                        pop -= count;
                    }

                    if pop == 0 {
                        break;
                    }
                }

                if pop != 0 {
                    println!("EMPTY");
                    return;
                }
            }
            "Top" => {
                if stack.is_empty() {
                    println!("EMPTY");
                    return;
                }
                let top = stack[stack.len() - 1].1;
                println!("{}", top);
            }
            "Size" => {
                println!("{}", size);
            }
            _ => unreachable!(),
        }
    }
    println!("SAFE");
}

pub struct IO<R, W: std::io::Write>(R, std::io::BufWriter<W>);

impl<R: std::io::Read, W: std::io::Write> IO<R, W> {
    pub fn new(r: R, w: W) -> Self {
        Self(r, std::io::BufWriter::new(w))
    }
    pub fn write<S: ToString>(&mut self, s: S) {
        use std::io::Write;
        self.1.write_all(s.to_string().as_bytes()).unwrap();
    }
    pub fn read<T: std::str::FromStr>(&mut self) -> T {
        use std::io::Read;
        let buf = self
            .0
            .by_ref()
            .bytes()
            .map(|b| b.unwrap())
            .skip_while(|&b| b == b' ' || b == b'\n' || b == b'\r' || b == b'\t')
            .take_while(|&b| b != b' ' && b != b'\n' && b != b'\r' && b != b'\t')
            .collect::<Vec<_>>();
        unsafe { std::str::from_utf8_unchecked(&buf) }
            .parse()
            .ok()
            .expect("Parse error.")
    }
    pub fn vec<T: std::str::FromStr>(&mut self, n: usize) -> Vec<T> {
        (0..n).map(|_| self.read()).collect()
    }
    pub fn chars(&mut self) -> Vec<char> {
        self.read::<String>().chars().collect()
    }
}

Submission Info

Judge Result