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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
use std::fs::File;
use std::io::Read;
#[derive(Debug, PartialEq, Eq, PartialOrd, Ord)]
struct Hand {
typ: u32,
cards: Vec<u8>,
bid: u32,
}
fn count_cards(arr: &Vec<u8>) -> Vec<u32> {
let mut count = [0; 256].to_vec();
for a in arr {
count[*a as usize] += 1;
}
let mut count = count
.iter()
.copied()
.filter(|x| *x != 0)
.collect::<Vec<_>>();
count.sort();
count.reverse();
count
}
fn card_type(count: &Vec<u32>) -> u32 {
// returns card type from card count
if count[0] == 5 {
return 6;
} else if count[0] == 4 {
return 5;
} else if count[0..2] == [3, 2] {
return 4;
} else if count[0] == 3 {
return 3;
} else if count[0..2] == [2, 2] {
return 2;
} else if count[0] == 2 {
return 1;
} else {
return 0;
}
}
fn get_card_type1(c: &Vec<u8>) -> u32 {
card_type(&count_cards(&c))
}
fn get_card_type2(c: &Vec<u8>) -> u32 {
let c_noj: Vec<u8> = c.iter().copied().filter(|x| *x != 0).collect();
let joker = c
.iter()
.fold(0, |acc, x| if *x == 0 { acc + 1 } else { acc });
let mut count = count_cards(&c_noj);
count.push(0);
count[0] += joker;
card_type(&count)
}
fn string_to_card(s: &str, map: &[char]) -> Vec<u8> {
s.chars()
.map(|c| map.iter().position(|x| x == &c).unwrap() as u8)
.collect()
}
fn main() {
// let filename = "in/day07.ref";
let filename = "in/day07.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();
let map1 = [
'2', '3', '4', '5', '6', '7', '8', '9', 'T', 'J', 'Q', 'K', 'A',
];
let map2 = [
'J', '2', '3', '4', '5', '6', '7', '8', '9', 'T', 'Q', 'K', 'A',
];
let mut hands1: Vec<Hand> = Vec::new();
let mut hands2: Vec<Hand> = Vec::new();
for line in lines.iter() {
let (cards, bid) = line.split_once(" ").unwrap();
let cards1 = string_to_card(&cards, &map1);
let cards2 = string_to_card(&cards, &map2);
hands1.push(Hand {
cards: cards1.clone(),
bid: bid.parse().unwrap(),
typ: get_card_type1(&cards1),
});
hands2.push(Hand {
cards: cards2.clone(),
bid: bid.parse().unwrap(),
typ: get_card_type2(&cards2),
});
}
hands1.sort();
let mut res1 = 0;
for (i, h) in hands1.iter().enumerate() {
// println!("{:?} {}", h.cards, h.typ);
res1 += (i as u32 + 1) * h.bid;
}
hands2.sort();
let mut res2 = 0;
for (i, h) in hands2.iter().enumerate() {
// println!("{:?} {}", h.cards, h.typ);
res2 += (i as u32 + 1) * h.bid;
}
println!("res1: {}", res1);
println!("res2: {}", res2);
assert_eq!(res1, 250232501);
assert_eq!(res2, 249138943);
}
|
