use aoc_runner_derive::{aoc, aoc_generator};

type Data = [usize;9];

#[aoc_generator(day06)]
pub fn input_generator(input: &str) -> [usize;9] {
    let mut state: [usize;9] = [0;9];
    let input = input.trim().split(',').map(|v| v.parse::<usize>().unwrap());
    for v in input {
        state[v]+=1;
    }
    state
}


fn simulate(input: &Data,n: usize) -> usize {
    let mut state=*input;
    for i in 0..n {
        state[(7+i)%9]+=state[i%9];
    }
    state.iter().sum()

}

#[aoc(day06, part1)] // 353274
pub fn solve_part1(input: &Data) -> usize {
    simulate(input,80)
}

#[aoc(day06, part2)] // 1609314870967
pub fn solve_part2(input: &Data) -> usize {
    simulate(input,256)
}