blob: bd9f06ce5180e5fc50c7684af2b9563dc9fdacc2 (
plain)
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
|
use std::fs;
fn parse_input(input_file_path: &str, data: &mut Vec<Vec<i32>>) {
let input =
fs::read_to_string(input_file_path).expect("Expected to parse string successfully!");
let mut i = 0;
for line in input.split('\n') {
if line.is_empty() {
return;
}
// println!("line: {}", line);
let items = line
.split(' ')
.map(|item| return item.trim().parse::<i32>().unwrap());
let mut tmp: Vec<i32> = vec![];
for (j, item) in items.enumerate() {
tmp.insert(j, item);
}
data.insert(i, tmp);
i += 1;
}
}
fn is_safe(arr: &[i32]) -> bool {
let mut is_increasing = true;
let mut is_decreasing = true;
for i in 0..arr.len() - 1 {
let diff = arr[i] - arr[i + 1];
if diff.abs() < 1 || diff.abs() > 3 {
return false;
}
if arr[i] - arr[i + 1] > 0 {
is_increasing = false;
}
if arr[i] - arr[i + 1] < 0 {
is_decreasing = false;
}
}
is_increasing || is_decreasing
}
fn is_also_safe(arr: &[i32]) -> bool {
let safe = is_safe(arr);
let mut also_safe = false;
if !safe {
for (i, _item) in arr.iter().enumerate() {
let mut test = arr.to_vec();
test.remove(i);
also_safe = if is_safe(&test) { true } else { also_safe };
}
}
safe || also_safe
}
pub fn part1(input_file_path: &str) {
println!("Part 1, Day 2");
let mut data = Vec::new();
parse_input(input_file_path, &mut data);
let mut c = 0;
for line in data {
if is_safe(&line) {
c += 1;
}
}
println!("Safe lines: {}", c);
}
pub fn part2(input_file_path: &str) {
println!("Part 2, Day 2");
let mut data = Vec::new();
parse_input(input_file_path, &mut data);
let mut c = 0;
for line in data {
if is_also_safe(&line) {
c += 1;
}
}
println!("Safe lines: {}", c);
}
|
