結果

問題 No.967 引き算をして門松列(その2)
ユーザー noriocnorioc
提出日時 2024-08-14 12:55:52
言語 Elixir
(1.18.1)
結果
AC  
実行時間 1,125 ms / 2,000 ms
コード長 1,271 bytes
コンパイル時間 1,029 ms
コンパイル使用メモリ 63,280 KB
実行使用メモリ 59,468 KB
最終ジャッジ日時 2024-08-14 12:56:06
合計ジャッジ時間 12,094 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
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
    s = IO.binread(:eof) |> String.split("\n", trim: true)
    # s = File.read!("/dev/stdin") |> String.split("\n", trim: true)
    for t <- tl(s) do
      # [a, b, c, x, y, z] = li()
      [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

  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 ->
        x = b - a + 1
        f(a, b-x, c)
      b <= c ->
        x = c - b + 1
        f(a, b, c-x)
      true ->
        [{a, c, b}, {b, c, a}, {b, a, c}, {c, a, b}]
    end
  end
end
0