考虑下面的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
解法
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
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;
}