day21 done

1 files changed, 11 insertions, 7 deletions

diff --git a/src/2015/day21/aoc.cpp b/src/2015/day21/aoc.cpp
index 8c7e259..e9b303a 100644
--- a/src/2015/day21/aoc.cpp
+++ b/src/2015/day21/aoc.cpp
@@ -2,22 +2,26 @@
 
 namespace aoc2015 {
 
-int day21() {
+std::pair<int, int> day21() {
   Role me{100, 0, 0, 0};
   Role boss{100, 8, 2, 0};
   RPG game;
 
-  int money{INT32_MAX};
+  int m1{INT32_MAX};
+  int m2{INT32_MIN};
   std::vector<Role> rs = game.shop(me);
-  std::for_each(rs.begin(), rs.end(), [&game, &boss, &money](Role& r) {
+  std::for_each(rs.begin(), rs.end(), [&game, &boss, &m1, &m2](Role& r) {
     Role b = boss;
     if (game.battle(r, b)) {
-      if (r.cost < money) {
-        money = r.cost;
+      if (r.cost < m1) {
+        m1 = r.cost;
+      }
+    } else {
+      if (r.cost > m2) {
+        m2 = r.cost;
       }
     }
   });
-  return money;
+  return {m1, m2};
 }
-
 } // namespace aoc2015