path: root/2015/rs/src/six.rs

static INPUT: &'static str = include_str!("six.txt");

#[derive(Debug)]
enum LightCommand {
  TurnOff,
  TurnOn,
  Toggle
}

fn main() {
  let valueified = INPUT.split('\n').filter(|s| !s.is_empty()).map(|s| {
    let mut rsplit = s.rsplit(' ');
    let mut cd = rsplit.next().unwrap().split(',').map(str::parse::<usize>).map(Result::unwrap);
    let mut ab = rsplit.nth(1).unwrap().split(',').map(str::parse::<usize>).map(Result::unwrap);
    let ab = (ab.next().unwrap(), ab.next().unwrap());
    let cd = (cd.next().unwrap(), cd.next().unwrap());
    match &s[0..7] {
      "turn on" => (LightCommand::TurnOn, ab,cd),
      "toggle " => (LightCommand::Toggle, ab,cd),
      "turn of" => (LightCommand::TurnOff, ab,cd),
      _ => unreachable!()
    }
  });

  let mut lights = vec![false; 1_000_000];
  for (cmd, (x1, y1), (x2, y2)) in valueified.clone() {
    use LightCommand::*;
    for x in x1..=x2 {
      for y in y1..=y2 {
        match cmd {
          TurnOff => { lights[x + y * 1000] = false; },
          TurnOn => { lights[x + y * 1000] = true; },
          Toggle => { lights[x + y * 1000] = !lights[x + y * 1000]; }
        }
      }
    }
  }
  let light_count = lights.into_iter().filter(|a| *a).count();
  println!("Lit light count: {light_count}");

  let mut lights = vec![0u64; 1_000_000];
  for (cmd, (x1, y1), (x2, y2)) in valueified {
    use LightCommand::*;
    for x in x1..=x2 {
      for y in y1..=y2 {
        match cmd {
          TurnOff if lights[x + y * 1000] != 0 => { lights[x + y * 1000] -= 1; },
          TurnOn => { lights[x + y * 1000] += 1; },
          Toggle => { lights[x + y * 1000] += 2 },
          _ => ()
        }
      }
    }
  }
  let light_count = lights.into_iter().sum::<u64>();
  println!("Lit light count(Ancient Nordic Elvish): {light_count}");
}