fizz-buzz-whatever.R

# https://en.wikipedia.org/wiki/Fizz_buzz

values <- 1:15

rules <- c("fizz" = 3, "buzz" = 5)

container <- rep("", length.out = length(values))

for (name in names(rules)) {
  rule_idx <- values %% rules[name] == 0
  container <- ifelse(rule_idx, paste0(container, name), container)
}

blank_idx <- container == ""
container <- ifelse(blank_idx, values, container)

# Level Up -----------------------------------------------
values2 <- 1:105
rules2 <- c("fizz" = 3, "buzz" = 5, "pop" = 7)

soda_shop <- function(rules, values) {
  container <- rep("", length.out = length(values))
  
  for (name in names(rules)) {
    idx <- values %% rules[name] == 0
    container <- ifelse(idx, paste0(container, name), container)
  }
  
  idx <- container == ""
  ifelse(idx, values, container)
}

soda_shop(rules2, 1:105)