Solve p111 in rust

_data/answers.tsv

@@ -70,6 +70,7 @@ ID	type	size	answer
 92	int	32	8581146
 97	int	64	8739992577
 99	int	16	709
+111	uint	64	0
 118	uint	16	44680
 123	int	16	21035
 134	int	64	18613426663617118

rust/src/include/problems.rs

@@ -18,6 +18,7 @@ use crate::p0069::p0069;
 use crate::p0076::p0076;
 use crate::p0077::p0077;
 use crate::p0087::p0087;
+use crate::p0111::p0111;
 use crate::p0187::p0187;
 use crate::p0357::p0357;
 use crate::p0836::p0836;
@@ -50,6 +51,7 @@ seq!(N in 0001..=0020 {
             76 =>  Some(( &76, p0076, false)),
             77 =>  Some(( &77, p0077, false)),
             87 =>  Some(( &87, p0087, false)),
+            111 => Some((&111, p0111, false)),
             187 => Some((&187, p0187, false)),
             357 => Some((&357, p0357, true)),
             836 => Some((&836, p0836, false)),

rust/src/lib.rs

@@ -23,6 +23,7 @@ pub mod p0069;
 pub mod p0076;
 pub mod p0077;
 pub mod p0087;
+pub mod p0111;
 pub mod p0187;
 pub mod p0357;
 pub mod p0836;

rust/src/main.rs

@@ -40,6 +40,7 @@ pub mod p0069;
 pub mod p0076;
 pub mod p0077;
 pub mod p0087;
+pub mod p0111;
 pub mod p0187;
 pub mod p0357;
 pub mod p0836;
@@ -97,6 +98,7 @@ seq!(N in 01..=20 {
 #[case::problem_76(76)]
 #[case::problem_77(77)]
 #[case::problem_87(87)]
+#[case::problem_111(111)]
 #[case::problem_187(187)]
 // #[case::problem_357(357)]
 #[case::problem_836(836)]

rust/src/p0111.rs

@@ -0,0 +1,30 @@
+/*
+Project Euler Problem 111
+
+*/
+use std::iter::zip;
+
+use crate::include::primes::primes_until;
+use crate::include::utils::Answer;
+
+pub fn p0111() -> Answer {
+    let ten_9: u64 = 10_u64.pow(9);
+    let ten_10: u64 = 10_u64.pow(10);
+    let mut answer: u64 = 0;
+    let mut subanswer: Vec<u64> = vec![0; 10];
+    let mut current: Vec<usize> = vec![0; 10];
+    for p in primes_until::<u64>(ten_10).filter(|x| *x > ten_9) {
+        let s = p.to_string();
+        for digit in 0..=9 {
+            let idx = digit as usize;
+            let count = s.bytes().filter(|b| *b == digit + b'0').count();
+            if count > current[idx] {
+                current[idx] = count;
+                subanswer[idx] = p;
+            } else if count == current[idx] {
+                subanswer[idx] += p;
+            }
+        }
+    }
+    return Answer::Int(subanswer.into_iter().sum::<u64>().into());
+}