contest 158

1221. Split a String in Balanced Strings/main.py

@@ -0,0 +1,12 @@
+class Solution:
+    def balancedStringSplit(self, s: str) -> int:
+        res = 0
+        flag = 0
+        for c in s:
+            if c == 'L':
+                flag += 1
+            else:
+                flag -= 1
+            if flag == 0:
+                res += 1
+        return res

1222. Queens That Can Attack the King/main.py

@@ -0,0 +1,18 @@
+class Solution:
+    def queensAttacktheKing(self, queens: List[List[int]], king: List[int]) -> List[List[int]]:
+        res = []
+        dirs = [(0, 1), (0, -1), (1, 0), (-1, 0),
+                (1, 1), (-1, -1), (1, -1), (-1, 1)]
+        queens_set = set(tuple(queen) for queen in queens)
+        for d in dirs:
+            x, y = king
+            while True:
+                x, y = x + d[0], y + d[1]
+                if 0 <= x < 8 and 0 <= y < 8:
+                    if (x, y) in queens_set:
+                        res.append([x, y])
+                        break
+                else:
+                    break
+
+        return res