2016/src/bin/day19.rs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

use std::collections::VecDeque;
use std::io::{self, Read};

#[derive(Clone, Copy)]
struct Elf {
    position: usize,
    gifts: usize,
}

fn solve(count: usize) -> usize {
    let mut circle: VecDeque<Elf> = (0..count)
        .map(|i| Elf { position: i + 1, gifts: 1 })
        .collect();

    while circle.len() > 1 {
        let mut current = circle.pop_front().unwrap();
        let next = circle.pop_front().unwrap();
        current.gifts += next.gifts;
        circle.push_back(current);
    }

    circle.pop_front().unwrap().position
}

fn main() {
    let mut input = String::new();
    io::stdin().read_to_string(&mut input).unwrap();

    println!("Part 1: {}", solve(input.trim().parse().unwrap()));
}

#[test]
fn part1() {
    assert_eq!(3, solve(5));
}