finish day6

Author: BWbwchen

day6/Cargo.lock

@@ -0,0 +1,8 @@
+[package]
+name = "day6"
+version = "0.1.0"
+edition = "2021"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]

day6/Cargo.toml

@@ -0,0 +1 @@
+1,1,1,2,1,1,2,1,1,1,5,1,1,1,1,1,1,1,1,1,1,2,1,1,1,1,1,4,1,1,1,1,3,1,1,3,1,1,1,4,1,5,1,3,1,1,1,1,1,5,1,1,1,1,1,5,5,2,5,1,1,2,1,1,1,1,3,4,1,1,1,1,1,1,1,1,1,1,1,1,2,1,1,1,1,5,4,1,1,1,1,1,5,1,2,4,1,1,1,1,1,3,3,2,1,1,4,1,1,5,5,1,1,1,1,1,2,5,1,4,1,1,1,1,1,1,2,1,1,5,2,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,4,3,1,1,3,1,3,1,4,1,5,4,1,1,2,1,1,5,1,1,1,1,1,5,1,1,1,1,1,1,1,1,1,4,1,1,4,1,1,1,1,1,1,1,5,4,1,2,1,1,1,1,1,1,1,1,1,1,1,3,1,1,1,1,1,1,1,1,1,1,4,1,1,1,2,1,4,1,1,1,1,1,1,1,1,1,4,2,1,2,1,1,4,1,1,1,1,1,1,3,1,1,1,1,1,1,1,1,3,2,1,4,1,5,1,1,1,4,5,1,1,1,1,1,1,5,1,1,5,1,2,1,1,2,4,1,1,2,1,5,5,3

day6/input/1.txt

@@ -0,0 +1 @@
+3,4,3,1,2

day6/input/2.txt

@@ -0,0 +1,15 @@
+mod part1;
+use part1::part1;
+
+fn main() {
+    let s_input = include_str!("../input/1.txt")
+        .split("\n")
+        .filter(|&n| !n.is_empty())
+        .collect::<String>();
+    let input = s_input
+        .split(",")
+        .map(|n| n.parse::<u64>().unwrap())
+        .collect::<Vec<_>>();
+    part1(&input, 80);
+    part1(&input, 256);
+}

day6/src/main.rs

@@ -0,0 +1,14 @@
+pub fn part1(e: &Vec<u64>, day: i32) {
+    let mut dp = vec![0_u64; 9];
+
+    e.iter().for_each(|&n| dp[n as usize] += 1);
+
+    (0..day).for_each(|_| {
+        dp.rotate_left(1);
+        dp[6] += dp[8];
+    });
+
+    let ans = dp.iter().sum::<u64>();
+
+    println!("{}", ans);
+}