結果

問題 No.2247 01 ZigZag
ユーザー siman
提出日時 2023-03-18 16:20:42
言語 Ruby
(3.4.1)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 1,331 bytes
コンパイル時間 211 ms
コンパイル使用メモリ 7,296 KB
実行使用メモリ 14,720 KB
最終ジャッジ日時 2024-09-18 16:15:30
合計ジャッジ時間 7,516 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 49 WA * 1
権限があれば一括ダウンロードができます
コンパイルメッセージ
Main.rb:87: warning: ambiguous first argument; put parentheses or a space even after `-' operator
Syntax OK

ソースコード

diff # 

N, M, K = gets.split.map(&:to_i)

def f(n, m, k)
  zero_stack = []
  one_stack = []

  (k + 1).times do |i|
    if i.even?
      zero_stack << 1
    else
      one_stack << 1
    end
  end

  if zero_stack.size > n
    return false
  end

  if one_stack.size > m
    return false
  end

  zero_stack[0] += n - zero_stack.size if zero_stack.size > 0
  one_stack[-1] += m - one_stack.size if one_stack.size > 0

  res = ''
  i = 0

  while zero_stack.size > 0 || one_stack.size > 0
    if i.even?
      res << '0' * zero_stack.shift
    else
      res << '1' * one_stack.shift
    end

    i += 1
  end

  res
end

def g(n, m, k)
  zero_stack = []
  one_stack = []

  (k + 1).times do |i|
    if i.even?
      one_stack << 1
    else
      zero_stack << 1
    end
  end

  if zero_stack.size > n
    return false
  end

  if one_stack.size > m
    return false
  end

  zero_stack[0] += n - zero_stack.size if zero_stack.size > 0
  one_stack[-1] += m - one_stack.size if one_stack.size > 0

  res = ''
  i = 0

  while zero_stack.size > 0 || one_stack.size > 0
    if i.even?
      res << '1' * one_stack.shift
    else
      res << '0' * zero_stack.shift
    end

    i += 1 end

  res
end

res1 = f(N, M, K)
res2 = g(N, M, K)
ans = []
ans << res1 if res1
ans << res2 if res2

if ans.empty?
  puts -1
else
  puts ans.sort.first
end
0