minor cleanup on day1, use Iterator::sum/product

This commit is contained in:
Stephen Waits committed 2020-12-02 22:28:38 -08:00
1 parent 86ac6625de
commit 3ab5c7926d
1 file changed
+6 -4
+6 -4
View File
@@ -5,8 +5,9 @@ use anyhow::{anyhow, Result};
fn part1(data: &[i64]) -> Result<i64> {
for c in utils::combinations(2, data.len()) {
if data[c[0]] + data[c[1]] == 2020 {
return Ok(data[c[0]] * data[c[1]]);
let test = vec![data[c[0]], data[c[1]]];
if test.iter().sum::<i64>() == 2020 {
return Ok(test.iter().product());
}
}
Err(anyhow!("no result found"))
@@ -14,8 +15,9 @@ fn part1(data: &[i64]) -> Result<i64> {
fn part2(data: &[i64]) -> Result<i64> {
for c in utils::combinations(3, data.len()) {
if data[c[0]] + data[c[1]] + data[c[2]] == 2020 {
return Ok(data[c[0]] * data[c[1]] * data[c[2]]);
let test = vec![data[c[0]], data[c[1]], data[c[2]]];
if test.iter().sum::<i64>() == 2020 {
return Ok(test.iter().product());
}
}
Err(anyhow!("no result found"))