leetcode - 1647. Minimum Deletions to Make Character Frequencies Unique

Description

A string s is called good if there are no two different characters in s that have the same frequency.

Given a string s, return the minimum number of characters you need to delete to make s good.

The frequency of a character in a string is the number of times it appears in the string. For example, in the string “aab”, the frequency of ‘a’ is 2, while the frequency of ‘b’ is 1.

Example 1:

Input: s = "aab"
Output: 0
Explanation: s is already good.

Example 2:

Input: s = "aaabbbcc"
Output: 2
Explanation: You can delete two 'b's resulting in the good string "aaabcc".
Another way it to delete one 'b' and one 'c' resulting in the good string "aaabbc".

Example 3:

Input: s = "ceabaacb"
Output: 2
Explanation: You can delete both 'c's resulting in the good string "eabaab".
Note that we only care about characters that are still in the string at the end (i.e. frequency of 0 is ignored).

Constraints:

1 <= s.length <= 10^5
s contains only lowercase English letters.

Solution

Greedy, first delete the most frequent character, delete until the frequency is unique.

Time complexity: $o(n)$?
Space complexity: $o(n)$

Code

class Solution:
    def minDeletions(self, s: str) -> int:
        fre_cnt = {}
        for c in s:
            fre_cnt[c] = fre_cnt.get(c, 0) + 1
        res = 0
        need_delete = []
        exists = []
        for c, v in fre_cnt.items():
            if v in exists:
                need_delete.append(v)
            else:
                exists.append(v)
        exists.sort()
        need_delete.sort()
        for i in need_delete:
            new_i = i - 1
            while new_i > 0 and bisect.bisect_left(exists, new_i) < len(exists) and exists[bisect.bisect_left(exists, new_i)] == new_i:
                new_i -= 1
            new_i_index = bisect.bisect_left(exists, new_i)
            exists.insert(new_i_index, new_i)
            res += i - new_i
        return res

Or a simpler one:

class Solution:
    def minDeletions(self, s: str) -> int:
        fre_cnt = {}
        for c in s:
            fre_cnt[c] = fre_cnt.get(c, 0) + 1
        res = 0
        exists = set()
        for c, v in fre_cnt.items():
            while v > 0 and v in exists:
                res += 1
                v -= 1
            exists.add(v)
        return res