From: Ramakrishnan Muthukrishnan <ram@rkrishnan.org>
Date: Mon, 29 Dec 2014 07:02:21 +0000 (+0530)
Subject: hw12: solution to exercise 2
X-Git-Url: https://git.rkrishnan.org/provisioning?a=commitdiff_plain;h=8dcef3b4c30b76c85cf6052e43d82fc7d84fa2b6;p=yorgey.git

hw12: solution to exercise 2
---

diff --git a/hw12/Risk.hs b/hw12/Risk.hs
index b78ac10..965c7c7 100644
--- a/hw12/Risk.hs
+++ b/hw12/Risk.hs
@@ -5,7 +5,7 @@ module Risk where
 import Control.Applicative
 import Control.Monad.Random
 import Control.Monad(replicateM, liftM)
-import Data.List(sort)
+import Data.List(sortBy)
 
 ------------------------------------------------------------
 -- Die values
@@ -30,10 +30,17 @@ type Army = Int
 
 data Battlefield = Battlefield { attackers :: Army, defenders :: Army }
 
--- battle :: Battlefield -> Rand StdGen Battlefield
+newBattlefield :: Battlefield -> [DieValue] -> [DieValue] -> Battlefield
+newBattlefield battlefield rollA rollB =
+    let pairList = zip (sortBy (flip compare) rollA) (sortBy (flip compare) rollB)
+    in
+      foldr (\(a, d) acc -> if a > d then Battlefield (attackers acc) ((defenders acc) - 1)
+                            else Battlefield ((attackers acc) - 1)  (defenders acc)) battlefield pairList
+
+
+battle :: Battlefield -> Rand StdGen Battlefield
 battle (Battlefield {attackers = a, defenders = d}) =
   let rolla = replicateM (min (a - 1) 3) die
       rolld = replicateM (min d 2) die
-      zipped = liftA2 zip rolla rolld
   in
-   
+    liftA2 (newBattlefield (Battlefield a d)) rolla rolld