diff --git a/1221. Split a String in Balanced Strings/main.py b/1221. Split a String in Balanced Strings/main.py new file mode 100644 index 0000000..fb3bd7c --- /dev/null +++ b/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 diff --git a/1222. Queens That Can Attack the King/main.py b/1222. Queens That Can Attack the King/main.py new file mode 100644 index 0000000..3718bca --- /dev/null +++ b/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