No Such Blog or Diary

ICPC2005 Regional, Tokyo の問題Ｂを Haskell で．queue を使ったシミュレーションだけど面倒だから一ターンごとにリスト生成...

 -- Problem B in ACM/ICPC 2005 ASIA Regional Tokyo
 -- 2005/11/04     Brute Force
import Control.Monad
import Debug.Trace
main = getProblems >>= mapM_ (putStrLn.show.solve)
  
getProblems = 
    do
    [m, c, n] <- getNums
    if (n==0 && m==0 && c==0) then return []
       else do
            xs <- replicateM n getEntry
            liftM ((m,c,n,xs):) getProblems
    where 
    getEntry = getNums >> getNums
    getNums = liftM (map read.words) getLine
  
 -- it's better to make each entry of ds the pair of it and its length 
solve (m,c,n,xs) = sl xs 0 (take m $ repeat [])
    where
    sl [] t _ = t
    sl ys t ds = let
                 (hs, ys') = unzip $ map (\x->(head x, tail x)) ys
                 (t', ds') = sl' ds hs
                 in sl (filter (not.(==[])) ys') (t'+t) ds'
    sl' ds = foldl searchOne (0,ds)
    searchOne (t, ds) x = let 
                          p = length $ takeWhile (not.or.map (==x)) ds 
                          ds' =if p<m then take p ds++[filter (not.(==x)) (ds!!p)]++drop (p+1) ds
                               else ds  
                          in insertOne (t+p+1) ds' x
    insertOne t ds x =
        if length (head ds) < c then (t+1, (x:head ds):tail ds)
        else let
             p = length $ takeWhile ((==c).length) ds 
             ds' = if p<m then take p ds++[x:(ds!!p)]++drop (p+1) ds
                   else ds
             p' = length $ takeWhile ((==c).length) ds'
             hds = head ds
             tds = tail ds
             q = p'-1
             tds' =if p'<m then take q tds++[last hds:(tds!!q)]++drop (q+1) tds
                   else tds
             in (t+p+p'+p+5, (x:init hds):tds')

Microsoft から来たメールによると，どうやら theSpokeアカウント を使わずに「Visual Studio 2005 Professional」を手に入れられるようになるらしい．いろいろと文句や問い合わせが行った結果なのかどうかしらんけどありがたいかも．

ASCII24の記事によると『Visual Studio .NET the Spoke Premium 2003』を購入、登録した学生を対象に『Visual Studio 2005 Professional Edition』を追加費用なしでダウンロード提供してくれるらしい．詳しい情報を得ようとだいぶ前に登録したきりの"the Spoke”のサイトへ行ってみるが...　３ヶ月間ログインしないと登録が抹消されますって言われて登録消えているらしい orz．　もう一回登録しなおしてもいいものなのかどうか...

終クロ６上巻６００ページ弱，下巻６００ページ超え．なんだかもう記録に挑戦？　同時に買ったシャナの５冊分って...　とりあえず７巻の１２５０円が何ページになるか楽しみだ，と．(つーか，文庫本のカバーがかかるのだろうか？)

Pitts の論文でも読むか．

ICPCをHaskellでやろうとすると入出力はこんな感じだろうか？　とりあえずメインルーチンはデータセットのリストを getProblems で生成し，mapM_ で各データセットに対して solve で解いた結果を putStrLn.show で出力と．

main = getProblems >>= mapM_ (putStrLn.show.solve)

getProblems に関して．各入力セットの先頭の数字が終了フラグになるタイプは，終了なら [] をリターン，そうでなければデータをタプルにして cons をリフトする．

main = getProblems >>= mapM_ (putStrLn.show.solve)
getProblems = 
    do
    [n] <- getNums
    if n==0 then return []
       else do
            xs <- replicateM n getEntry
            liftM (xs:) getProblems
    where 
    getEntry = liftM words getLine
    getNums = liftM (map read.words) getLine
 

入力の最初にデータセットの数がある場合は replicateM でリストにすると．

main = getProblems >>= mapM_ (putStrLn.show.solve)
getProblems = 
    do
    n <- liftM (head.map read.words) getLine
    replicateM n getProblem
getProblem =
    do
    r <- getNum
    s <- liftM (head.map read.words) getLine
    ws <- replicateM s getNum
    return (r, ws)
    where 
    getNum = liftM (head.map read.words) getLine :: IO(Double)

あとは EOF で終了という不親切な場合もあるが...　面倒なのでやめとこう．各データセット内の値の取得は，replicateM と liftM (map read.words) getLine と叫べば大体は入力できそう．面倒なのは read の型を明示しとかないとこける可能性があることだろう．