-
Notifications
You must be signed in to change notification settings - Fork 0
/
P1373.rs
69 lines (63 loc) · 2.05 KB
/
P1373.rs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
/*
Author : quickn (quickn.ga)
Email : [email protected]
*/
use std::str;
use std::io::{self, BufWriter, Write};
/* https://github.com/EbTech/rust-algorithms */
/// Same API as Scanner but nearly twice as fast, using horribly unsafe dark arts
/// **REQUIRES** Rust 1.34 or higher
pub struct UnsafeScanner<R> {
reader: R,
buf_str: Vec<u8>,
buf_iter: str::SplitAsciiWhitespace<'static>,
}
impl<R: io::BufRead> UnsafeScanner<R> {
pub fn new(reader: R) -> Self {
Self {
reader,
buf_str: Vec::new(),
buf_iter: "".split_ascii_whitespace(),
}
}
/// This function should be marked unsafe, but noone has time for that in a
/// programming contest. Use at your own risk!
pub fn token<T: str::FromStr>(&mut self) -> T {
loop {
if let Some(token) = self.buf_iter.next() {
return token.parse().ok().expect("Failed parse");
}
self.buf_str.clear();
self.reader
.read_until(b'\n', &mut self.buf_str)
.expect("Failed read");
self.buf_iter = unsafe {
let slice = str::from_utf8_unchecked(&self.buf_str);
std::mem::transmute(slice.split_ascii_whitespace())
}
}
}
}
fn main() {
let (stdin, stdout) = (io::stdin(), io::stdout());
let (mut scan, mut sout) = (UnsafeScanner::new(stdin.lock()), BufWriter::new(stdout.lock()));
// Big-endian
let s: String = scan.token();
let mut s_chars: Vec<char> = s.chars().collect();
// Little-endian
s_chars.reverse();
let mut res: String = String::new();
let mut tmp = 0;
for i in 0..s_chars.len() {
let t = (s_chars[i] as u8) - ('0' as u8);
tmp += t*((1 << (i % 3)) as u8);
if i % 3 == 2 {
res.push_str(tmp.to_string().as_str());
tmp = 0;
}
}
if tmp != 0 || res.len() == 0 {
res.push_str(tmp.to_string().as_str());
}
writeln!(sout, "{}", res.chars().rev().collect::<String>()).ok();
}