DEV Community

Colin Fay
Colin Fay

Posted on • Originally published at colinfay.me on

Advent of Code 2019-07 with R

Solving Advent of Code 2019-07 with R.

[Disclaimer] Obviously, this post contains a big spoiler about Adventof Code, as it gives solutions for solving day 7.

Instructions

Find the instructions at: https://adventofcode.com/2019/day/7

R solution

vec <- scan("input7.txt", numeric(), sep = ",")

# Getting parse_opcode  
source("intcode_helpers.R")

prog <- function(vec, inputs, pos = 1){ 
  ipt_stat <- 1
  stock <- 1
  while (pos < length(vec)) {

    c(op_code, one, two, three) %<-% parse_opcode( vec, pos )

    if (op_code == 99) {
      return(
        list(
          vec = vec, 
          res = res, 
          pos = pos, 
          code = 99, 
          processed = FALSE
        )
      )
    }

    if (op_code %in% 1:2){

      if (op_code == 1) fun <- `+`
      if (op_code == 2) fun <- `*`
      vec[three] <- fun(one, two)
      pos <- pos + 4
    } else if (op_code == 3){
      if (stock > length(inputs)){
        return(
          list(
            vec = vec, 
            res = res, 
            pos = pos, 
            code = op_code, 
            processed = TRUE
          )
        )
      }
      if (ipt_stat == 1){
        vec[vec[pos + 1] + 1 ] <- inputs[1]
        ipt_stat <- 2
      } else {
        vec[vec[pos + 1] + 1 ] <- inputs[2]
      }
      pos <- pos + 2
      stock <- stock + 1
    } else if (op_code == 4){
      res <- vec[vec[pos + 1] + 1]
      pos <- pos + 2
    } else if (op_code == 5){
      if (one != 0){
        pos <- two + 1
      } else {
        pos <- pos + 3
      }
    } else if (op_code == 6){
      if (one == 0){
        pos <- two + 1
      } else {
        pos <- pos + 3
      }
    } else if (op_code == 7){
      if (one < two){
        vec[three] <- 1 
      } else {
        vec[three] <- 0 
      }
      pos <- pos + 4
    } else if (op_code == 8){
      if (one == two){
        vec[three] <- 1 
      } else {
        vec[three] <- 0 
      }
      pos <- pos + 4
    }
  }
  return(
    list(
      vec = vec, 
      res = res, 
      pos = pos, 
      code = op_code, 
      processed = FALSE
    )
  )
}

res <- 0

comb <- expand.grid(0:4, 0:4, 0:4, 0:4, 0:4) 

to_keep <- apply(comb, 1, function(x){
  length(unique(x)) != length(x)
})

comb <- comb[!to_keep,]
for (i in seq_len(nrow(comb))){
  phase <- comb[i,] %>% as.numeric()
  start <- 0
  for (j in seq_along(phase)){
    start <- prog(vec, c(phase[j], start))$res
  }
  res[i] <- start
}
max(res)


## [1] 67023

Enter fullscreen mode Exit fullscreen mode

Part two

library(purrr)


## 
## Attaching package: 'purrr'

## The following object is masked from 'package:magrittr':
## 
## set_names


res <- 0

comb <- expand.grid(5:9, 5:9, 5:9, 5:9, 5:9) 

to_keep <- apply(comb, 1, function(x){
  length(unique(x)) != length(x)
})
comb <- comb[!to_keep,]

for (i in seq_len(nrow(comb))){
  phase <- comb[i,] %>% as.numeric()

  A <- prog(vec, c(phase[1], 0))
  B <- prog(vec, c(phase[2], A$res))
  C <- prog(vec, c(phase[3], B$res))
  D <- prog(vec, c(phase[4], C$res))
  E <- prog(vec, c(phase[5], D$res))

  while(
    map_lgl(
      list(A, B, C, D, E), 
      "processed"
    ) %>% all()
  ){
    A <- prog(A$vec, E$res, pos = A$pos)
    B <- prog(B$vec, A$res, pos = B$pos)
    C <- prog(C$vec, B$res, pos = C$pos)
    D <- prog(D$vec, C$res, pos = D$pos)
    E <- prog(E$vec, D$res, pos = E$pos)
  }

  res[i] <- E$res
}
max(res)


## [1] 7818398

Enter fullscreen mode Exit fullscreen mode

Top comments (0)