blob: 0bed2c753782fdec677be463a316334518cb8158 (
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
|
#![allow(dead_code)]
#![allow(unused_variables)]
#![allow(unused_mut)]
use std::fs::File;
use std::io::Read;
fn main() {
// let filename = "in/day04.ref";
let filename = "in/day04.pzl";
let mut f = File::open(filename).expect("cannot open file");
let mut content = String::new();
f.read_to_string(&mut content).expect("cannot read file");
let lines: Vec<_> = content.trim_end().split('\n').collect();
// println!("{:?}", lines);
let mut res1 = 0;
let mut counts: Vec<i32> = Vec::new();
for line in &lines {
let s1: Vec<&str> = line.split(":").collect();
let game_num: u32 = s1[0]
.replace(" ", " ")
.replace(" ", " ")
.split(" ")
.collect::<Vec<&str>>()[1]
.parse()
.unwrap();
let s2: Vec<&str> = s1[1].trim().split("|").collect::<Vec<&str>>();
let winning: Vec<u32> = s2[0]
.trim()
.replace(" ", " ")
.split(" ")
.map(|x| x.parse::<u32>().unwrap())
.collect();
let actual: Vec<u32> = s2[1]
.trim()
.replace(" ", " ")
.replace(" ", " ")
.split(" ")
.map(|x| x.parse::<u32>().unwrap())
.collect();
let count: i32 = actual.iter().filter(|x| winning.contains(x)).count() as i32;
counts.push(count);
// println!("{}", count);
res1 += 1_i32.checked_shl((count - 1) as u32).unwrap_or(0);
}
let mut copies: Vec<i32> = vec![0; counts.len()];
for i in 0..counts.len() {
for j in i..i + counts[i] as usize {
copies[j + 1] += 1 + copies[i];
}
}
let res2 = copies.iter().sum::<i32>() + copies.len() as i32;
println!("res1: {}", res1);
println!("res2: {}", res2);
assert_eq!(res1, 21919);
assert_eq!(res2, 9881048);
}
|
