信息解码（Message Decoding, ACM/ICPC World Finals 1991, UVa 213）rust解法

考虑下面的01串序列：

0, 00, 01, 10, 000, 001, 010, 011, 100, 101, 110, 0000, 0001, ..., 1101, 1110, 00000, ...

首先是长度为1的串，然后是长度为2的串，依此类推。如果看成二进制，相同长度的后一个串等于前一个串加1。注意上述序列中不存在全为1的串。

你的任务是编写一个解码程序。首先输入一个编码头（例如AB#TANCnrtXc），则上述序列的每个串依次对应编码头的每个字符。例如，0对应A，00对应B，01对应#，...，110对应X，0000对应c。接下来是编码文本（可能由多行组成，你应当把它们拼成一个长长的01串）。编码文本由多个小节组成，每个小节的前3个数字代表小节中每个编码的长度（用二进制表示，例如010代表长度为2），然后是各个字符的编码，以全1结束（例如，编码长度为2的小节以11结束）。编码文本以编码长度为000的小节结束。

例如，编码头为$#**\，编码文本为0100000101101100011100101000，应这样解码：

010(编码长度为2)00(#)00(#)10(*)11(小节结束)011(编码长度为3)000()111(小节结束)001(编码长度为1)0($)1(小节结束)000(编码结束)。

样例

输入

$#**\
0100000101101100011100101000
TNM AEIOU   
0010101100011    
1010001001110110011   
11000

输出

##*\$  
TAN ME

解法

rust 复制代码

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

fn main() {
    let mut buf = String::new();
    io::stdin().read_line(&mut buf).unwrap();
    let mut kv = HashMap::new();
    let v: Vec<_> = buf.trim().chars().collect();
    let mut w = 0;
    let mut cnt = 0;
    'foo: loop {
        w += 1;
        for i in 0..((1<<w) - 1){
            let s = format!("{:0width$b}", i, width=w);
            //println!("{}", s);

            kv.insert(s, v[cnt]);
            cnt += 1;
            if cnt >= v.len() {
                break 'foo;
            }
        }
    }

    let mut buf = String::new();
    while let Ok(_size @ 1.. ) = io::stdin().read_line(&mut buf) {//读取多行
        buf = buf.trim().to_string();
    }
    let mut codes = buf.trim().to_string();
    while codes.len() > 0 {
        let s: String = codes.drain(0..3).collect();
        let mut bw:usize = 0;
        for c in s.chars() {
            bw *= 2;
            bw += c.to_digit(10).unwrap() as usize;
        }
        //println!("{}", bw);
        while codes.len() > 0 {
            let s: String = codes.drain(0..bw).collect();
            if let Some(c) = kv.get(&s) {
                print!("{}", c);
            } else {
                break;
            }
        }
    }
}

解法2

rust 复制代码

use std::{io};

fn main() {
    let mut buf = String::new();
    io::stdin().read_line(&mut buf).unwrap();
    let v: Vec<_> = buf.trim().chars().collect();

    let mut buf = String::new();
    while let Ok(_size @ 1.. ) = io::stdin().read_line(&mut buf) {//读取多行
        buf = buf.trim().to_string();
    }
    let mut codes = buf.trim().to_string();
    while codes.len() > 0 {
        let len = readint(&mut codes, 3);
        while len > 0 {
            let num = readint(&mut codes, len);
            if num == (1 << len) - 1 {//判断二进制编码是否全为1
                break;
            } else {
                let idx = (1 << len) - len + num - 1;//通过len和num可以唯一确定是第几个二进制编码
                print!("{}", v[idx]);
            }
        }
    }
}

fn readint(codes: &mut String, len: usize) -> usize {
    let s: String = codes.drain(0..len).collect();
    let mut num: usize = 0;
    for c in s.chars() {
        num *= 2;
        num += c.to_digit(10).unwrap() as usize;
    }
    return num;
}