Skip to content

Instantly share code, notes, and snippets.

@nobsun
Last active March 5, 2017 07:58
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save nobsun/12af13976157858fe6765da9bdf45b6e to your computer and use it in GitHub Desktop.
Save nobsun/12af13976157858fe6765da9bdf45b6e to your computer and use it in GitHub Desktop.
import Control.Arrow
import Data.Char
type Field = ([Int],Int)
ifield :: Field
ifield = (repeat 0,0)
data Piece = I | L | O | S | T deriving (Eq,Show,Read)
type Pos = Int
type Drop = (Pos,Piece)
update :: Field -> Drop -> Field
update cs (pos,pc) = dispatch pc cs pos
where
dispatch pc = case pc of
I -> updateI
L -> updateL
O -> updateO
S -> updateS
T -> updateT
updateI,updateL,updateO,updateS,updateT :: Field -> Pos -> Field
updateI (cs,hx) pc = case splitAt pc cs of
(xs,y:zs) -> (xs++y':zs,hx')
where
y' = y+4
hx' = hx `max` y'
updateL (cs,hx) pc = case splitAt pc cs of
(xs,y:z:zs) -> (xs++y':z':zs,hx')
where
y' = (y `max` z) + 3
z' = y' - 2
hx' = hx `max` y'
updateO (cs,hx) pc = case splitAt pc cs of
(xs,y:z:zs) -> (xs++y':z':zs,hx')
where
y' = (y `max` z) + 2
z' = y'
hx' = hx `max` y'
updateS (cs,hx) pc = case splitAt pc cs of
(xs,y:z:w:ws) -> (xs++y':z':w':ws,hx')
where
(y',z',w') = if t > w-1 then (t+1,t+2,t+2) else (w,w+1,w+1)
t = y `max` z
hx' = hx `max` w'
updateT (cs,hx) pc = case splitAt pc cs of
(xs,y:z:w:ws) -> (xs++y':z':w':ws,hx')
where
(y',z',w') = if t > z+1 then (t+1,t+1,t+1) else (z+2,z+2,z+2)
t = y `max` w
hx' = hx `max` y'
readDrops :: String -> [Drop]
readDrops "" = []
readDrops xs = case span isDigit xs of
(pos,pc:rs) -> (read pos,read [pc]) : readDrops rs
pile :: Field -> [Drop] -> Field
pile = foldl update
type Problem = String
type Answer = String
e12 :: Problem -> Answer
e12 = show . snd . pile ifield . readDrops
type Test = (Problem, Answer)
test :: Test -> Bool
test = uncurry (==) . (e12 *** id)
{- ^
>>> test ("1O3L0I0T", "5")
True
>>> test ("0I", "4")
True
>>> test ("0I0I", "8")
True
>>> test ("0I1I2I3I4I", "4")
True
>>> test ("0S0I", "5")
True
>>> test ("0I0S", "6")
True
>>> test ("2S0T2O3I", "8")
True
>>> test ("4O4T1T0S4L1L3L", "10")
True
>>> test ("0S2S4S6S8S10S12S14S", "16")
True
>>> test ("14S12S10S8S6S4S2S0S", "2")
True
>>> test ("5I2O10I0O4L10T9T11L8I2I10I12O7L12T12T12S11T9O10O13I12O10O7I9I7O0S1O2S0L1L", "23")
True
>>> test ("9T14L10L8T4I1T3S5I8T12O3S7L9O7L14T2I7O3S6S2L0L13T10O4I9T7L8S0I12O9S11L11T14T", "27")
True
>>> test ("9S9S7O11O16I2T9O12L10T9O0O13I9O1I2T14S7O9S11T5L7I14T13O0T12I3S10L10O7I15I6S2L12S8I16I3L", "23")
True
>>> test ("11T13I16S15T7O10L12S1I5I8S5I13I15O8S9I1T12I1S5S0L14I12L16T2S2S8L2S14L16O4I13L15L13S11S9T13S9S3L6O", "22")
True
>>> test ("12L10S7I5L14T12S9L1T14I0I5L1T2O18T9L0I15I16L10S1O15I0L17O5L18T4I18L7L7I13I3I12I2S3T5T3S16L14S14O11O15T14S", "17")
True
>>> test ("0S18S2S19I14T7L14L2L6I9I0L4I5L13L15I8S8T2I5I7O18T3S1T7I2L8O0S20T9I14T5L5I1T4L9O8T19T5S12O16T19L4O10O10T14L", "24")
True
>>> test ("7T5L6S4S8T6S10I19O20L14I18L21S7I11S11O1L13T20O9I7L2T8L2S20L3O14L9T17I8L8S14I6T2O11T21O18O6T15T1S3L6O19S18O20S19O16T6S14T", "26")
True
>>> test ("18S2I4S16L13S17I21O8I17T8I14O12T20I20S19S16S13T12T20I22I15O2I2I8I2S18I9I9T6O13O13L17I2L20L2L4I9I19O11T3S10O2S18T12I5O11S19O21S6I17T17S", "26")
True
>>> test ("11L5S0T22S18O13T2O22S15I12I21T16I3I1I22L11L11L22O13S24S15L13T15S19L10O15T7S24T19L0T13O11I12T13S4I24L15O3S19O10L19O0S20L7O11L21I22S18T19T23O8I22S24L0S", "21")
True
>>> test ("7L7I11T7S18O17L8S15L9I3O24S3O1O5O14L9T13S2O25S22T10T8L24S18S13T1O1L6I10I4S13O3S7L10T1T4L17S20I18O15S25S23S21I19T6O24S9L2O2O15L12L8L8O18I18L0T5O", "31")
True
>>> test ("999I999I999I999I999I999I999I999I999I999I999I", "44")
True
-}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment