Personal tools

Programming performance/Magnus Haskell

From HaskellWiki

< Programming performance(Difference between revisions)
Jump to: navigation, search
(mention minor bugs)
Line 39: Line 39:
           print result
           print result
'''Bugs''':  Two minor bugs here.  Purchases are made at ''lastPrice'' rather than ''todayPrice''.  The wrong column of the data is used.  It should be the last column not the 4th column.  [[User:Newsham|Newsham]] 17:53, 7 March 2007 (UTC)

Latest revision as of 17:53, 7 March 2007

  • Language: Haskell
  • Skill: Intermediate. I'm self taught in Haskell and use it as a hobbyist but haven't used it much for serious work.
  • Time: 25 minutes.
  • Notes: 10 minutes spent on a silly bug where I had typed 1 instead of 100. This code is not very pretty but it works. A while after submitting I noticed I used the wrong data column. It should say column 4, not 3 in closingPrice.

[edit] Code

import Data.List (partition,foldl')
closingPrice line =
    read (words line !! 3)
sell assets price = sum (map (\(_,shares)->shares*price) assets)
tradeOnce (cash,assets,lastPrice) todayPrice =
    if todayPrice <= lastPrice * 97/100 then
        let buyCost = cash*10/100 in
           (cash-buyCost, (lastPrice,buyCost/lastPrice):assets, todayPrice)
    else let (toSell,toKeep) =
                 partition (\(buyPrice,_)->todayPrice >= buyPrice*106/100) assets
         in (cash + sell toSell todayPrice,
notComment ('#':rest) = False
notComment _ = True
trade input =
    let (firstPrice:restPrices) = reverse $
                                  map closingPrice $
                                  filter notComment $
                                  lines input
    in foldl' tradeOnce (10000.0,[],firstPrice) restPrices
main = do input <- readFile "gspc.txt"
          let (cash,toSell,lastPrice) = trade input
          let result = cash + sell toSell lastPrice
          print result

Bugs: Two minor bugs here. Purchases are made at lastPrice rather than todayPrice. The wrong column of the data is used. It should be the last column not the 4th column. Newsham 17:53, 7 March 2007 (UTC)