Submission #826157

---

Source Code Expand

class Box
  def initialize(r,w)
    @red = r
    @white = w
  end
  def putRed(t)
    @red += t
  end
  def putWhite(t)
    @white += t
  end
  def getRed
    if(@red > 0)then
      @red -= 1
      return 1
    else
      return 0
    end
  end
  def getWhite
    if(@white > 0)then
      @white -= 1
      return 1
    else
      return 0
    end
  end
  def chkRed
    return @red
  end
  def chkWhite
    return @white
  end
end

N,M = gets.split(/\s/).map {|s|s.to_i}
boxList = Array.new(N)
N.times do |i|
  if(i==0)then
    boxList[i] = Box.new(1,0)
  else
    boxList[i] = Box.new(0,1)
  end
end

# (1) 赤だけあるときは、カウントしない。移動は赤にする。
# (2) 赤と白があるときは、カウントする。移動は赤にする。
# (3) 白だけあるときは、カウントしない。移動は白にする。
cnt = 1
redidx = -1
M.times do |i|
  x,y = gets.split(/\s/).map {|s|s.to_i}
  if(boxList[x-1].chkRed==1 && boxList[x-1].chkWhite==0)then
    # (1)
    boxList[y-1].putRed(boxList[x-1].getRed)
    #puts "(1)"
  elsif(boxList[x-1].chkRed==1 && boxList[x-1].chkWhite > 0)then
    # (2)
    cnt += 1
    # redがあればredを移動し、なければwhiteを移動する
    boxList[y-1].putRed(boxList[x-1].getRed)
    redidx = x
    #puts "(2)"
  else
    # (3) + 同じ箱から赤も白も取り去った記録があれば、cntを減らす
    boxList[y-1].putWhite(boxList[x-1].getWhite)
    if(x==redidx && boxList[x-1].chkWhite==0)then
      cnt -= 1
    end
    #puts "(3)"
  end
end

puts cnt

Submission Info

Judge Result