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
use std::collections::HashMap;
use std::error::Error;
use std::io::{self, Read};
fn main() -> Result<(), Box<Error>> {
let mut input = String::new();
io::stdin().read_to_string(&mut input)?;
let output = solve_1(&input, "\n");
println!("{}", output);
Ok(())
}
#[allow(dead_code)]
fn solve_0(input: &str, delimiter: &str) -> String {
let sum: i32 = input
.trim()
.split(delimiter)
.map(|change| change.parse::<i32>().unwrap())
.sum();
sum.to_string()
}
#[test]
fn test_solve_0() {
assert_eq!(solve_0("+1, -2, +3, +1", ", "), "3");
}
fn solve_1(input: &str, delimiter: &str) -> String {
vec![0 as i32] // Start with 0 frequency
.into_iter()
.chain(
input
.trim()
.split(delimiter)
.map(|change| change.parse::<i32>().unwrap())
.cycle(),
)
.scan(0, |freq, change| {
*freq += change;
Some(*freq)
})
.scan(HashMap::new(), |history, freq| {
let count = history.entry(freq).or_insert(0);
*count += 1;
Some((freq, *count))
})
.filter(|(_, count)| *count > 1)
.map(|(freq, _)| freq)
.next()
.unwrap()
.to_string()
}
#[test]
fn test_solve_1() {
assert_eq!(solve_1("+1, -2, +3, +1", ", "), "2");
assert_eq!(solve_1("+1, -1", ", "), "0");
assert_eq!(solve_1("+3, +3, +4, -2, -4", ", "), "10");
assert_eq!(solve_1("-6, +3, +8, +5, -6", ", "), "5");
assert_eq!(solve_1("+7, +7, -2, -7, -4", ", "), "14");
}