forked from ngthanhtrung23/CompetitiveProgramming
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathB.rs
More file actions
60 lines (53 loc) · 1.57 KB
/
B.rs
File metadata and controls
60 lines (53 loc) · 1.57 KB
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
use std::cmp::{max, min};
/// Scanner class
/// Copied from https://codeforces.com/contest/1168/submission/54903799
///
/// Usage:
/// ```
/// let mut scanner = Scanner::default();
/// let n = scan.next::<i32>();
/// ```
#[derive(Default)]
struct Scanner {
buffer: Vec<String>
}
impl Scanner {
fn next<T: std::str::FromStr>(&mut self) -> T {
loop {
if let Some(token) = self.buffer.pop() {
return token.parse().ok().expect("Failed parsing input")
}
let mut input = String::new();
std::io::stdin().read_line(&mut input).expect("Failed reading input");
self.buffer = input.split_whitespace().rev().map(String::from).collect();
}
}
}
fn main() {
let mut scanner = Scanner::default();
let n_query = scanner.next::<u32>();
for _ in 0..n_query {
let n = scanner.next::<usize>();
let a: Vec<i64> = (0..n).map(|_| scanner.next()).collect();
let sum_all = a.iter().sum();
let prefix_sum:Vec<i64> = a.iter().scan(0, |sum, &x| { *sum += x; Some(*sum) }).collect::<Vec<_>>();
let mut max_subarr_sum:i64 = a[0];
let mut min_prefix_sum:i64 = 0;
for i in 0..n-1 {
let cur_sum = prefix_sum[i];
max_subarr_sum = max(max_subarr_sum, cur_sum - min_prefix_sum);
min_prefix_sum = min(min_prefix_sum, cur_sum);
}
let mut min_prefix_sum:i64 = a[0];
for i in 1..n {
let cur_sum = prefix_sum[i];
max_subarr_sum = max(max_subarr_sum, cur_sum - min_prefix_sum);
min_prefix_sum = min(min_prefix_sum, cur_sum);
}
if max_subarr_sum < sum_all {
println!("YES");
} else {
println!("NO");
}
}
}