LeetCode/Python/2423.py

# 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())