Submission #17259514

---

Source Code Expand

defmodule Main do
  def main do
    [_n, m] =
      IO.read(:line) |> String.trim() |> String.split(" ") |> Enum.map(&String.to_integer/1)

    IO.read(:line)
    |> String.trim()
    |> String.split(" ")
    |> Enum.map(&String.to_integer/1)
    |> solve(m)
    |> IO.puts()
  end

  @doc ~S"""
  https://atcoder.jp/contests/abc161/tasks/abc161_b

  ## Examples

      iex> Abc161B.solve([5, 4, 2, 1], 1)
      "Yes"
      iex> Abc161B.solve([380, 19, 1], 2)
      "No"
      iex> Abc161B.solve([4, 56, 78, 901, 2, 345, 67, 890, 123, 45, 6, 789], 3)
      "Yes"

  """
  def solve(list, m) do
    sum = Enum.sum(list)

    Enum.sort(list, :desc)
    |> Enum.take(m)
    |> Enum.all?(fn a ->
      a * 4 * m >= sum
    end)
    |> if(do: "Yes", else: "No")
  end
end

Submission Info

Judge Result