Самые большие расходы

Часть раздела Логика и потоки выполнения путешествия по R на Coddy. Урок 60 из 64.

Задание

Средне

Добавьте команду TOP. Все команды из предыдущих шагов должны продолжать работать точно так же, как раньше.

TOP <n>: выведите n самых крупных расходов, по одному в строке, начиная с самого крупного, в формате 2024-05-02 rent 500.00. Расходы с одинаковой суммой перечисляются по дате, начиная с самой ранней. Если существует менее n расходов, выведите их все; если расходов нет, выведите no expenses.

Предоставленный код передаёт каждую строку входных данных в 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))
      }
    }
  }
  replies
}

# Предоставленный код ввода/вывода: оставьте его как есть
input <- suppressWarnings(readLines(file("stdin")))
cat(tracker(input), sep = "\n")