AC P1052

Author: eytter48

P1052.cpp

@@ -0,0 +1,37 @@
+#include "header.h"
+
+
+class Solution {
+public:
+    int maxSatisfied(vector<int>& customers, vector<int>& grumpy, int X) {
+        int hap = 0;
+        for (int i=0; i<customers.size(); i++) {
+            if (grumpy[i]==0) hap+=customers[i];
+        }
+        int win,h,t;
+        h=0; t=X; win=0;
+        for (int i=h; i<t && i<customers.size(); i++) {
+            if (grumpy[i]==1) win+=customers[i];
+        }
+        int maxWin = win;
+        while (t<customers.size()) {
+            if (grumpy[t]==1) win+=customers[t];
+            if (grumpy[h]==1) win-=customers[h];
+            t++; h++;
+            if (win>maxWin) maxWin = win;  
+        }
+        return hap+maxWin;
+    }
+};
+
+
+
+int main() {
+    vector<int> customers, grumpy;
+    int X;
+    customers = {1,0,1,2,1,1,7,5};
+    grumpy    = {0,1,0,1,0,1,0,1};
+    X = 3;
+    cout << Solution().maxSatisfied(customers, grumpy, X) << endl;
+    return 0;
+}