[LeetCode]1338. Reduce Array Size to The Half

1338. Reduce Array Size to The Half

Medium

You are given an integer array arr. You can choose a set of integers and remove all the occurrences of these integers in the array.

Return the minimum size of the set so that at least half of the integers of the array are removed.

 

Example 1:

Input: arr = [3,3,3,3,5,5,5,2,2,7]
Output: 2
Explanation: Choosing {3,7} will make the new array [5,5,5,2,2] which has size 5 (i.e equal to half of the size of the old array).
Possible sets of size 2 are {3,5},{3,2},{5,2}.
Choosing set {2,7} is not possible as it will make the new array [3,3,3,3,5,5,5] which has a size greater than half of the size of the old array.

Example 2:

Input: arr = [7,7,7,7,7,7]
Output: 1
Explanation: The only possible set you can choose is {7}. This will make the new array empty.

 

Constraints:

• 2 <= arr.length <= 105
• arr.length is even.
• 1 <= arr[i] <= 105

---

문제 풀이

1. 문제 접근
2. 리스트의 최대 길이는 10^5 이다.
3. 시간복잡도 O(N^2) 이하의 속도로 풀어야한다.

풀이

1. 숫자 갯수를 세는 것은 for 문을 한번 진행해서 N 시간만에 구할 수 있다. 
2. 편하게 Counter 를 사용해서 각 숫자의 갯수를 저장한다.
3. 가장 갯수가 많은 것부터 더 해가면서 그 값이 리스트 전체 길이의 반 이상이 되면 정답이 된다. 
4. 최소의 숫자 세트를 빼서 총 리스트 길이를 반 이상 줄여야 하기 때문에 정렬을 하여 큰 숫자부터 빼며 답을 찾는다.

소스 코드

class Solution:
    def minSetSize(self, arr: List[int]) -> int:
        cnt = Counter(arr)
        total_len = 0
        ans = 0
        n = len(arr)
        cnt = sorted(cnt.values(), reverse=True)
        for num in cnt:
            print(num)
            total_len += num
            ans += 1
            if total_len >= n - total_len:
                return ans