R/2-2.R

In soniamitchell/adventofcode: My Advent of Code solutions

#' @rdname day2
#' @param dat dat
#' @export
#'
dive2 <- function(dat) {
  # Initialise variables
  aim <- 0
  horizontal_position <- 0
  depth <- 0

  # Track aim
  for (i in seq_len(nrow(dat))) {
    if (dat$direction[i] == "down") {
      aim <- aim  + dat$value[i]
    } else if (dat$direction[i] == "up") {
      aim <- aim  - dat$value[i]
    } else {
      horizontal_position <- horizontal_position + dat$value[i]
      depth <- depth + (aim * dat$value[i])
    }
  }

  horizontal_position * depth
}