-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathP1149.rs
82 lines (74 loc) · 2.33 KB
/
P1149.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
70
71
72
73
74
75
76
77
78
79
80
81
82
/*
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())
}
}
}
}
use std::cmp::min;
const INF: u32 = 1_000_000_000;
fn main() {
let (stdin, stdout) = (io::stdin(), io::stdout());
let (mut scan, mut sout) = (UnsafeScanner::new(stdin.lock()), BufWriter::new(stdout.lock()));
let n: usize = scan.token();
let mut dp: Vec<Vec<u32>> = vec![vec![INF;3];n];
let (r_0, g_0, b_0): (u32, u32, u32) = (scan.token(), scan.token(), scan.token());
dp[0][0] = r_0;
dp[0][1] = g_0;
dp[0][2] = b_0;
for i in 1..n {
let (r, g, b): (u32, u32, u32) = (scan.token(), scan.token(), scan.token());
// R
for j in 0..3 {
if j != 0 {
dp[i][0] = min(dp[i][0], dp[i-1][j] + r);
}
}
// G
for j in 0..3 {
if j != 1 {
dp[i][1] = min(dp[i][1], dp[i-1][j] + g);
}
}
// R
for j in 0..3 {
if j != 2 {
dp[i][2] = min(dp[i][2], dp[i-1][j] + b);
}
}
}
writeln!(sout, "{}", min(dp[n-1][0], min(dp[n-1][1], dp[n-1][2]))).ok();
}