程序模拟（Concurrency Simulator, ACM/ICPC World Finals 1991, UVa210）rust解法

你的任务是模拟n个程序（按输入顺序编号为1～n）的并行执行。每个程序包含不超过25条语句，格式一共有5种：var = constant（赋值）；print var（打印）；lock；unlock；end。

变量用单个小写字母表示，初始为0，为所有程序公有（因此在一个程序里对某个变量赋值可能会影响另一个程序）。常数是小于100的非负整数。

每个时刻只能有一个程序处于运行态，其他程序均处于等待态。上述5种语句分别需要t1、t2、t3、t4、t5单位时间。运行态的程序每次最多运行Q个单位时间（称为配额）。当一个程序的配额用完之后，把当前语句（如果存在）执行完之后该程序会被插入一个等待队列中，然后处理器从队首取出一个程序继续执行。初始等待队列包含按输入顺序排列的各个程序，但由于lock/unlock语句的出现，这个顺序可能会改变。

lock的作用是申请对所有变量的独占访问。lock和unlock总是成对出现，并且不会嵌套。lock总是在unlock的前面。当一个程序成功执行完lock指令之后，其他程序一旦试图执行lock指令，就会马上被放到一个所谓的阻止队列的尾部（没有用完的配额就浪费了）。当unlock执行完毕后，阻止队列的第一个程序进入等待队列的首部。

输入n, t1, t2, t3, t4, t5, Q以及n个程序，按照时间顺序输出所有print语句的程序编号和结果。

样例：

输入

复制代码

3 1 1 1 1 1 1
a = 4
print a
lock
b = 9
print b
unlock
print b
end
a = 3
print a
lock
b = 8
print b
unlock
print b
end
b = 5
a = 17
print a
print b
lock
b = 21
print b
unlock
print b
end

输出

复制代码

解法：

rust 复制代码

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

fn main() {
    let mut buf = String::new();
    io::stdin().read_line(&mut buf).unwrap();
    let v: Vec<usize> = buf.split_whitespace().map(|x| x.parse().unwrap()).collect();
    let n = v[0];
    let q = v[6];
    let mut programs = vec![];
    for _i in 0..n {
        let mut prog_stats = vec![];
        loop {
            let mut buf = String::new();
            io::stdin().read_line(&mut buf).unwrap();
            let mut run_time = 0;
            if let Some(_idx) = buf.find('=') {
                run_time = v[1];
            } else if buf.starts_with("print") {
                run_time = v[2];
            } else if buf.starts_with("lock") {
                run_time = v[3];
            } else if buf.starts_with("unlock") {
                run_time = v[4];
            } else if buf.starts_with("end") {
                run_time = v[5];
            }
            prog_stats.push((buf.trim().to_string(), run_time));
            if buf.trim() == "end" {
                programs.push(prog_stats);
                break;
            }
        }
    }
    let mut wait_progs = VecDeque::new();
    for i in 0..n {
        wait_progs.push_back((i, 0, 0));
    }
    let mut block_progs = VecDeque::new();
    let mut lock = (false, 0);
    let mut vars = HashMap::new();
    while let Some(mut prg) = wait_progs.pop_front() {
        let mut run_time = q;
        while run_time > 0 {
            let statement = &programs[prg.0][prg.1].0;
            if statement.starts_with("lock") {
                if lock.0 == true && lock.1 != prg.0 {
                    block_progs.push_back(prg);
                    break;
                } else {
                    lock.0 = true;
                    lock.1 = prg.0;
                }
            } else if statement.starts_with("unlock") {
                lock.0 = false;
                lock.1 = 0;
                if let Some(prg) = block_progs.pop_front() {
                    wait_progs.push_front(prg);
                }
            } else if let Some(idx) = statement.find('=') {
                let var = statement[0..idx - 1].to_string();
                let value: usize = statement[idx + 2..].parse().unwrap();
                vars.insert(var, value );
            } else if statement.starts_with("print") {
                let var = statement[6..].to_string();
                println!("{}: {}", prg.0 + 1, vars.get(&var).unwrap());
            }
            let min_time = (programs[prg.0][prg.1].1 - prg.2).min(run_time);
            prg.2 += min_time;
            run_time -= min_time;
            if prg.2 < programs[prg.0][prg.1].1 {
                wait_progs.push_back(prg);
            } else if prg.1 < programs[prg.0].len() - 1 {
                prg.1 += 1;
                prg.2 = 0;
                if run_time == 0 {
                    wait_progs.push_back(prg);
                }
            } else {
                break;
            }
        }
    }
}