Off-by-one bug fix

This commit is contained in:
2018-12-14 12:02:27 +01:00
parent 5c4646b323
commit ccdbea4a1c

View File

@@ -45,13 +45,13 @@ fn find_first(n: usize, len: usize) -> usize {
if result >= 10 { if result >= 10 {
current = update_current(current, result as usize / 10, mod_base); current = update_current(current, result as usize / 10, mod_base);
if current == n { if current == n {
return state.len() - 5; return state.len() - len + 1;
} }
state.push(result / 10); state.push(result / 10);
} }
current = update_current(current, result as usize % 10, mod_base); current = update_current(current, result as usize % 10, mod_base);
if current == n { if current == n {
return state.len() - 5 + 1; return state.len() - len + 1;
} }
state.push(result % 10); state.push(result % 10);