ボールカウント(野球)

hena3pre.rb

#!/usr/bin/env ruby
#http://qiita.com/Nabetani/items/ebd8a56b41711ba459f9
#STDOUT.sync=true
def solve(l)
    r=[]
    a=[0,0,0]
    l.chars{|e|
        a[1]+=1 if e=='s'
        a[2]+=1 if e=='b'
        a[1]+=1 if e=='f'&&a[1]<2
        (a[0]+=1;a[1]=a[2]=0) if e=='p'||a[1]>2
        a[1]=a[2]=0 if e=='h'||a[2]>3
        a[0]=a[1]=a[2]=0 if a[0]>2
        r<<a*''
    }
    r*','
end
DATA.each_with_index{|e,i|
    input,expect=e.chomp.split(' -> ')
    puts 'Case '+(i+1).to_s+': '+(expect==solve(input)?'OK':'NG')
}
__END__
s -> 010
sss -> 010,020,100
bbbb -> 001,002,003,000
ssbbbb -> 010,020,021,022,023,000
hsbhfhbh -> 000,010,011,000,010,000,001,000
psbpfpbp -> 100,110,111,200,210,000,001,100
ppp -> 100,200,000
ffffs -> 010,020,020,020,100
ssspfffs -> 010,020,100,200,210,220,220,000
bbbsfbppp -> 001,002,003,013,023,000,100,200,000
sssbbbbsbhsbppp -> 010,020,100,101,102,103,100,110,111,100,110,111,200,000,100
ssffpffssp -> 010,020,020,020,100,110,120,200,210,000