Create remove-colored-pieces-if-both-neighbors-are-the-same-color.py

Author: kamyu104

Python/remove-colored-pieces-if-both-neighbors-are-the-same-color.py

@@ -0,0 +1,18 @@
+# Time:  O(n)
+# Space: O(1)
+
+class Solution(object):
+    def winnerOfGame(self, colors):
+        """
+        :type colors: str
+        :rtype: bool
+        """
+        cnt1 = cnt2 = 0
+        for i in xrange(1, len(colors)-1):
+            if not (colors[i-1] == colors[i] == colors[i+1]):
+                continue
+            if colors[i] == 'A':
+                cnt1 += 1
+            else:
+                cnt2 += 1
+        return cnt1 > cnt2