当前位置：首页 > article >正文

铁轨（Rails, ACM/ICPC CERC 1997, UVa 514）rust解法

article 2024/9/24 23:32:19

有一个火车站，铁轨铺设如图6-1所示。有n节车厢从A方向驶入车站，按进站顺序编号为1～n。你的任务是判断是否能让它们按照某种特定的顺序进入B方向的铁轨并驶出车站。例如，出栈顺序(5 4 1 2 3)是不可能的，但(5 4 3 2 1)是可能的。
在这里插入图片描述

为了重组车厢，你可以借助中转站C。这是一个可以停放任意多节车厢的车站，但由于末端封顶，驶入C的车厢必须按照相反的顺序驶出C。对于每个车厢，一旦从A移入C，就不能再回到A了；一旦从C移入B，就不能回到C了。换句话说，在任意时刻，只有两种选择：A→C和C→B。
【分析】
在中转站C中，车厢符合后进先出的原则，因此是一个栈。

样例：
输入

5
5 4 3 2 1

输出

yes

解法：

use std::{collections::VecDeque, io};

fn main() {
    let mut buf = String::new();
    io::stdin().read_line(&mut buf).unwrap();
    let n: usize = buf.trim().parse().unwrap();
    let mut source = VecDeque::from((1..=n).collect::<Vec<_>>());
    let mut buf = String::new();
    io::stdin().read_line(&mut buf).unwrap();
    let target: Vec<usize> = buf.split_whitespace().map(|x| x.parse().unwrap()).collect();
    let mut b = 0;
    let mut s: VecDeque<usize> = VecDeque::new();
    while b < n {
        if let Some(&e) = source.front() {
            if e == target[b] {
                source.pop_front();
                b += 1;
            } else if let Some(&e) = s.back() {
                if e == target[b] {
                    s.pop_back();
                    b += 1;
                } else {
                    s.push_back(source.pop_front().unwrap());
                }
            } else {
                s.push_back(source.pop_front().unwrap());
            }
        } else if let Some(&e) = s.back() {
            if e == target[b] {
                b += 1;
                s.pop_back();
            } else {
                break;
            }
        }
    }
    if b == n {
        println!("yes");
    } else {
        println!("no");
    }
}