Got slightly confused on part 2... nearly there
This commit is contained in:
parent
f3c2f6cfd8
commit
6bd170cbde
@ -1,4 +1,4 @@
|
||||
use std::collections::HashMap;
|
||||
use std::{borrow::BorrowMut, collections::HashMap};
|
||||
|
||||
use super::Solution;
|
||||
|
||||
@ -16,40 +16,41 @@ impl Solution for Day04 {
|
||||
.collect();
|
||||
let mut ans = 0;
|
||||
|
||||
for cards in cards {
|
||||
let mut winning_numbers: HashMap<u32, bool> = HashMap::new();
|
||||
let mut sep_found = false;
|
||||
let mut card_nums = vec![];
|
||||
|
||||
for num in cards.split_whitespace() {
|
||||
// Are we after the |
|
||||
if num == "|" {
|
||||
sep_found = true;
|
||||
continue;
|
||||
}
|
||||
let num = num.parse::<u32>()?;
|
||||
|
||||
if sep_found {
|
||||
if winning_numbers.contains_key(&num) {
|
||||
card_nums.push(num);
|
||||
}
|
||||
} else {
|
||||
winning_numbers.insert(num, true);
|
||||
}
|
||||
}
|
||||
if !card_nums.is_empty() {
|
||||
ans += 2u32.pow((card_nums.len() - 1) as u32);
|
||||
for card in cards {
|
||||
let winning_nums = self.get_count_winning_numbers(card)?;
|
||||
if winning_nums > 0 {
|
||||
ans += 2u32.pow((winning_nums - 1) as u32);
|
||||
}
|
||||
}
|
||||
|
||||
Ok(Box::new(ans))
|
||||
}
|
||||
|
||||
fn part2(
|
||||
&self,
|
||||
_input: &mut Vec<String>,
|
||||
input: &mut Vec<String>,
|
||||
) -> Result<Box<dyn std::fmt::Display>, Box<dyn std::error::Error>> {
|
||||
Ok(Box::new("Ready"))
|
||||
// Remove Card XXX: at start and append card index
|
||||
let mut cards: Vec<(usize, &str)> = input
|
||||
.iter()
|
||||
.map(|c| c.split(":").into_iter().last().unwrap())
|
||||
.into_iter()
|
||||
.enumerate()
|
||||
.collect();
|
||||
|
||||
for card in cards {}
|
||||
// for (i, card) in cards.iter() {
|
||||
// let winning_nums = self.get_count_winning_numbers(card)?;
|
||||
// if winning_nums == 0 {
|
||||
// continue;
|
||||
// }
|
||||
// for c in cards.clone()[(i + 1)..=(i + 1 + winning_nums)].iter() {
|
||||
// cards.push(c.clone());
|
||||
// }
|
||||
// }
|
||||
// for i in cards.clone() {
|
||||
// println!("{:?}", i);
|
||||
// }
|
||||
Ok(Box::new(cards.len()))
|
||||
}
|
||||
|
||||
fn get_day(&self) -> u8 {
|
||||
@ -58,7 +59,7 @@ impl Solution for Day04 {
|
||||
}
|
||||
|
||||
impl Day04 {
|
||||
fn get_count_winning_numbers(card: &str) -> Result<usize, Box<dyn std::error::Error>> {
|
||||
fn get_count_winning_numbers(&self, card: &str) -> Result<usize, Box<dyn std::error::Error>> {
|
||||
let mut winning_numbers: HashMap<u32, bool> = HashMap::new();
|
||||
let mut sep_found = false;
|
||||
let mut card_nums = vec![];
|
||||
@ -120,6 +121,6 @@ mod test {
|
||||
.unwrap()
|
||||
.to_string();
|
||||
|
||||
assert_eq!(answer, "Ready");
|
||||
assert_eq!(answer, "30");
|
||||
}
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user