月間サマリー

CoddyのRジャーニー「ロジックとフロー」セクションの一部。レッスン 61/64。

チャレンジ

中級

MONTH コマンドを追加してください。これまでの手順にあったすべてのコマンドが、以前とまったく同じように動作し続けるようにしてください。

MONTH <YYYY-MM>：日付がその月で始まる支出のみを対象にします。支出の件数、合計（小数点以下2桁）、およびその月で最も金額の大きい支出のカテゴリ（同額のものが複数ある場合は最初のもの）を含めて、2024-05: count 3, total 520.50, biggest rent と返します。該当する支出がない場合は、2024-05: no expenses と返します。

提供されたコードは、入力行をそれぞれ1つのコマンドとして tracker() に渡し、返された各応答をそれぞれ別の行に出力します。日付は YYYY-MM-DD 形式で記述され、カテゴリは小文字の単語であり、入力にはここまでに説明したコマンドのみが含まれます。

自分で試してみよう

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)))
    } else if (parts[1] == "CATEGORY") {
      if (nrow(expenses) == 0) {
        replies <- c(replies, "no expenses")
      } else {
        totals <- tapply(expenses$amount, expenses$category, sum)
        replies <- c(replies, sprintf("%s: %.2f", names(totals), totals))
      }
    } else if (parts[1] == "TOP") {
      n <- as.numeric(parts[2])
      ranked <- expenses[order(-expenses$amount, expenses$date), ]
      top <- head(ranked, n)
      if (nrow(top) == 0) {
        replies <- c(replies, "no expenses")
      } else {
        replies <- c(replies, sprintf("%s %s %.2f", top$date, top$category, top$amount))
      }
    }
  }
  replies
}

# 提供された入出力コード: そのままにしておく
input <- suppressWarnings(readLines(file("stdin")))
cat(tracker(input), sep = "\n")