結果

問題 No.2247 01 ZigZag
ユーザー siman
提出日時 2023-03-24 16:39:23
言語 Ruby
(3.4.1)
結果
AC  
実行時間 166 ms / 2,000 ms
コード長 1,385 bytes
コンパイル時間 539 ms
コンパイル使用メモリ 7,552 KB
実行使用メモリ 14,848 KB
最終ジャッジ日時 2024-09-18 16:28:30
合計ジャッジ時間 7,509 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 50
権限があれば一括ダウンロードができます
コンパイルメッセージ
Main.rb:4: warning: ambiguous first argument; put parentheses or a space even after `-' operator
Main.rb:92: warning: ambiguous first argument; put parentheses or a space even after `-' operator
Syntax OK

ソースコード

diff # 

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

if K == 0 && (N != 0 && M != 0)
  puts -1
  exit
end

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