Альтернативный Нэш

lab2/prisoner/game.hs

@@ -21,13 +21,12 @@ nashEquilibriumStrategy opponentMoves generatedMoves n =
       nashEquilibriumStrategy (tail opponentMoves) (generatedMoves ++ [nextStep]) (n + 1)
     else generatedMoves
   where
-    cases = [['П', 'С'], ['П', 'П']]
-    results = [[0, 10], [5, 5]]
-    result =
-      [ min (results !! 0 !! 1) (results !! 1 !! 1),
-        max (results !! 0 !! 0) (results !! 1 !! 0)
-      ]
-    nextStep = cases !! indexOf results result !! 1
+    cases = [['С', 'С'], ['С', 'П'], ['П', 'С'], ['П', 'П']]
+    results = [[1, 1], [10, 0], [0, 10], [5, 5]]
+    p_years = min (results !! 1 !! 1) (results !! 3 !! 1)
+    s_years = min (results !! 0 !! 1) (results !! 2 !! 1)
+    nextStep | p_years <= s_years = 'П'
+             | otherwise = 'С'
 
 getScore :: [Char] -> [Char] -> Int -> Int -> (Int, Int)
 getScore moves1 moves2 score1 score2 =
@@ -47,7 +46,8 @@ game playerMoves gameStrategy
 
 
 playerMoves = ['С', 'П', 'С', 'С', 'П', 'П', 'С', 'С', 'П', 'П']
-gameStrategy = forgivingStrategy -- forgivingStrategy, nashEquilibriumStrategy
+-- gameStrategy = forgivingStrategy
+gameStrategy = nashEquilibriumStrategy
 computerMoves = game playerMoves gameStrategy
 (score1, score2) = getScore playerMoves computerMoves 0 0