day02

day02/main.cpp

@@ -53,21 +53,32 @@ int main(int argc, char **argv) {
     }
 
     string line;
-    int total = 0;
+    int total1 = 0;
+    int total2 = 0;
     while (getline(inputFile, line)) {
         bool valid = true;
         gameLine game = parseLine(line);
+        map<string, int> maxColor = {{"red", 0}, {"green", 0}, {"blue", 0}};
         for (auto &color : game.colors) {
             for (auto &c : color) {
                 if (gameBag[c.first] < c.second) {
                     valid = false;
                 }
+                if (maxColor[c.first] < c.second) {
+                    maxColor[c.first] = c.second;
+                }
             }
         }
         if (valid) {
             cout << "Game " << game.gameId << " is valid" << endl;
-            total += game.gameId;
+            total1 += game.gameId;
+        }
+        int power = 1;
+        for (auto &c : maxColor) {
+            power *= c.second;
         }
+        total2 += power;
     }
-    cout << "Total: " << total << endl;
+    cout << "Total1: " << total1 << endl;
+    cout << "Total2: " << total2 << endl;
 }