Wydatki według kategorii

Część sekcji Logika i przepływ sterowania ścieżki R w Coddy. Lekcja 59 z 64.

Wyzwanie

Łatwy

Dodaj polecenie CATEGORY. Wszystkie polecenia z poprzednich kroków mają nadal działać dokładnie tak jak wcześniej.

CATEGORY: odpowiedz jednym wierszem dla każdej kategorii, w kolejności alfabetycznej, food: 20.50, podając łączną sumę dla kategorii z dokładnością do dwóch miejsc po przecinku. Jeśli niczego nie zapisano, odpowiedz no expenses.

Jedno polecenie CATEGORY generuje kilka odpowiedzi, a każda z nich jest wyświetlana w osobnym wierszu.

Dostarczony kod przekazuje każdy wiersz wejścia do tracker() jako jedno polecenie i wyświetla każdą zwróconą odpowiedź w osobnym wierszu. Daty są zapisywane w formacie YYYY-MM-DD, kategorie to małe litery, a wejście zawiera tylko opisane dotąd polecenia.

Spróbuj swoich sił

tracker <- function(commands) {
  expenses <- data.frame(date = character(0), category = character(0), amount = numeric(0))
  replies <- c()
  for (cmd in commands) {
    parts <- strsplit(cmd, " ")[[1]]
    if (parts[1] == "ADD") {
      amount <- suppressWarnings(as.numeric(parts[4]))
      if (length(parts) != 4 || is.na(amount) || amount <= 0) {
        replies <- c(replies, paste("invalid:", cmd))
      } else {
        expenses <- rbind(expenses, data.frame(date = parts[2], category = parts[3], amount = amount))
        replies <- c(replies, sprintf("added %s %.2f", parts[3], amount))
      }
    } else if (parts[1] == "TOTAL") {
      replies <- c(replies, sprintf("total: %.2f, count: %d", sum(expenses$amount), nrow(expenses)))
    }
  }
  replies
}

# Dostarczony kod wejścia/wyjścia: pozostaw go bez zmian
input <- suppressWarnings(readLines(file("stdin")))
cat(tracker(input), sep = "\n")