-
Notifications
You must be signed in to change notification settings - Fork 0
/
P6236.rs
88 lines (81 loc) · 2.32 KB
/
P6236.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
83
84
85
86
87
88
/*
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::max;
fn main() {
let (stdin, stdout) = (io::stdin(), io::stdout());
let (mut scan, mut sout) = (UnsafeScanner::new(stdin.lock()), BufWriter::new(stdout.lock()));
let (n, m): (usize, usize) = (scan.token(), scan.token());
let mut daily = vec![0;n];
let mut max_v = 0;
for i in 0..n {
daily[i] = scan.token();
max_v = max(daily[i], max_v);
}
let (mut l, mut r) = (max_v, 1_000_000_000);
let f = |mid: usize| -> usize {
if max_v > mid {
return 0;
}
let mut func = 0;
let mut left = 0;
for i in 0..n {
if left < daily[i] {
left = mid;
func += 1;
}
left -= daily[i];
}
return func;
};
while l <= r {
let mid = (l+r)>>1;
let func = f(mid);
if func > m {
l = mid+1;
} else {
r = mid-1;
}
}
while l > 1 && f(l-1) == f(l) {
l -= 1;
}
writeln!(sout, "{}", l).ok();
}