結果

問題 No.967 引き算をして門松列(その2)
ユーザー norioc
提出日時 2024-08-14 18:40:20
言語 Elixir
(1.18.1)
結果
AC  
実行時間 1,142 ms / 2,000 ms
コード長 1,240 bytes
コンパイル時間 2,328 ms
コンパイル使用メモリ 63,192 KB
実行使用メモリ 61,796 KB
最終ジャッジ日時 2024-08-14 18:40:34
合計ジャッジ時間 14,002 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 10
権限があれば一括ダウンロードができます

ソースコード

diff #

defmodule Main do
  def input, do: IO.read(:line) |> String.trim
  def ii, do: input() |> String.to_integer
  def li, do: input() |> String.split |> Enum.map(&String.to_integer/1)
  def yn(b), do: IO.puts(if b, do: "Yes", else: "No")

  def main do
    # t = ii()
    # s = File.read!("/dev/stdin") |> String.split("\n", trim: true)
    s = IO.binread(:eof) |> String.split("\n", trim: true)
    for t <- tl(s) do
      [a, b, c, x, y, z] = t |> String.split |> Enum.map(&String.to_integer/1)
      case kadomatu(a, b, c) do
        [] -> -1
        res ->
          res
          |> Enum.map(fn {a1, b1, c1} -> x*(a-a1) + y*(b-b1) + z*(c-c1) end)
          |> Enum.min
      end
    end
    |> Enum.join("\n")
    |> IO.puts
  end

  @spec kadomatu(any(), any(), any()) :: list()
  def kadomatu(a, b, c) do
    for {a, b, c} <- [{b, a, c}, {b, c, a}, {a, c, b}, {c, a, b}] do
      f(a, b, c)
    end
    |> List.flatten
    |> Enum.uniq
    |> Enum.filter(fn {x, y, z} -> a >= x and b >= y and c >= z end)
  end

  def f(1, _, _), do: []
  def f(_, 1, _), do: []
  def f(a, b, c) do
    cond do
      a <= b -> f(a, a-1, c)
      b <= c -> f(a, b, b-1)
      true ->
        [{a, c, b}, {b, c, a}, {b, a, c}, {c, a, b}]
    end
  end
end
0