Przeglądaj źródła

Adding solution for 2423

Vinicius Teshima 1 miesiąc temu
rodzic
commit
a7f184529d
1 zmienionych plików z 44 dodań i 0 usunięć
  1. 44 0
      Python/2423.py

+ 44 - 0
Python/2423.py

@@ -0,0 +1,44 @@
+# 2423. Remove Letter To Equalize Frequency
+
+import sys
+
+class Solution:
+    def equalFrequency(self, word: str) -> bool:
+        def check_freq(word) -> bool:
+            l = [0 for _ in range(ord('z')+10)]
+            for c in word:
+                l[ord(c)] += 1
+                pass
+            n = 0
+            for x in l:
+                if x == 0:
+                    continue
+                if n == 0:
+                    n = x
+                    continue
+                if n != x:
+                    return False
+                pass
+            return True
+
+        for i in range(len(word)):
+            if check_freq(word[:i] + word[i+1:]):
+                return True
+            pass
+        return False
+    pass
+
+def main() -> int:
+    def r(word: str, exp: bool) -> None:
+        print(
+            f"Solution().equalFrequency({word}) = {Solution().equalFrequency(word)} | exp: {exp}"
+        )
+        pass
+
+    r('abcc', True)
+    r('aazz', False)
+    r('cccaa', True)
+    return 0
+
+if __name__ == "__main__":
+    sys.exit(main())