[2017][rust][6.x] rustfmt

sorbet
Alpha Chen 7 years ago
parent 2fb5e076e0
commit bb8f1c5da3

@ -1,20 +1,24 @@
use std::collections::HashSet; use std::collections::HashSet;
use failure::*; use failure::*;
#[allow(dead_code)]
pub fn solve(input: &str) -> Result<String, Error> { pub fn solve(input: &str) -> Result<String, Error> {
let banks: Vec<_> = input let banks: Vec<_> = input
.split_whitespace() .split_whitespace()
.map(str::parse) .map(str::parse)
.collect::<Result<_, _>>()?; .collect::<Result<_, _>>()?;
let mut reallocation = Reallocation { banks: banks.clone() }; let mut reallocation = Reallocation {
banks: banks.clone(),
};
let cycles = unique_cycles(reallocation); let cycles = unique_cycles(reallocation);
reallocation = Reallocation { banks }; reallocation = Reallocation { banks };
Ok((unique_cycles(reallocation.skip(cycles - 1)) - 1).to_string()) Ok((unique_cycles(reallocation.skip(cycles - 1)) - 1).to_string())
} }
fn unique_cycles<R: Iterator<Item=Vec<usize>>>(reallocation: R) -> usize { #[allow(dead_code)]
fn unique_cycles<R: Iterator<Item = Vec<usize>>>(reallocation: R) -> usize {
let mut seen = HashSet::new(); let mut seen = HashSet::new();
for (i, banks) in reallocation.enumerate() { for (i, banks) in reallocation.enumerate() {
if seen.contains(&banks) { if seen.contains(&banks) {
@ -28,10 +32,13 @@ fn unique_cycles<R: Iterator<Item=Vec<usize>>>(reallocation: R) -> usize {
#[test] #[test]
fn test_part_one() { fn test_part_one() {
let reallocation = Reallocation { banks: vec![0, 2, 7, 0] }; let reallocation = Reallocation {
banks: vec![0, 2, 7, 0],
};
assert_eq!(unique_cycles(reallocation), 5); assert_eq!(unique_cycles(reallocation), 5);
} }
#[allow(dead_code)]
struct Reallocation { struct Reallocation {
banks: Vec<usize>, banks: Vec<usize>,
} }

Loading…
Cancel
Save