[LeetCode] 2554. Maximum Number of Integers to Choose From a Range I (Python)

 

 

난이도: Medium

문제 설명

---

You are given an integer array banned and two integers n and maxSum. You are choosing some number of integers following the below rules:

• The chosen integers have to be in the range [1, n].
• Each integer can be chosen at most once.
• The chosen integers should not be in the array banned.
• The sum of the chosen integers should not exceed maxSum.

Return the maximum number of integers you can choose following the mentioned rules.

 

문제 예제

---

Example 1:

Input: banned = [1,6,5], n = 5, maxSum = 6
Output: 2
Explanation: You can choose the integers 2 and 4.
2 and 4 are from the range [1, 5], both did not appear in banned, and their sum is 6, which did not exceed maxSum.

 

Example 2:

Input: banned = [1,2,3,4,5,6,7], n = 8, maxSum = 1
Output: 0
Explanation: You cannot choose any integer while following the mentioned conditions.

Example 3:

Input: banned = [11], n = 7, maxSum = 50
Output: 7
Explanation: You can choose the integers 1, 2, 3, 4, 5, 6, and 7.
They are from the range [1, 7], all did not appear in banned, and their sum is 28, which did not exceed maxSum.

 

제한 사항

---

• 1 <= banned.length <= 10^4
• 1 <= banned[i], n <= 10^4
• 1 <= maxSum <= 10^9

 

✏️Solution(솔루션)

---

class Solution:
    def maxCount(self, banned: List[int], n: int, maxSum: int) -> int:

        check = defaultdict(bool)
        new_banned = []
        for banned_num in banned:
            if not check[banned_num]:
                new_banned.append(banned_num)
                check[banned_num] = True

        sum_num = 0
        cnt = 0
        for i in range(1, n+1):
            if i not in new_banned:
                sum_num += i
                cnt += 1
            if sum_num > maxSum:
                cnt -= 1
                break
        
        return cnt

 

 

[1, n] 범위에 있는 숫자들을 차례대로 보면서 banned에 없는 숫자이면 sum_num에 +i를 해주었고 cnt에는 +1을 해주었다.

 

그러다가 sum_num이 maxSum보다 커지게 되면 cnt를 다시 -1하고 반복문을 탈출 했고, cnt를 반환했다.

 

처음에는 단순히 위에 글 처럼 코드를 구현하였는데 마지막쯤 테스트케이스에서 시간초과가 나는 것을 확인했다. 

테스트케이스를 확인해보니 input 데이터가 다음과 같았다.

 

 

딱보니 banned에 중복되는 숫자가 들어가 banned에 길이가 커져서 if i not in banned: 코드에서 시간 초과가 나는 것 같았다.

 

그래서 중복을 제거하기 위해 new_banned를 만들었고 if i not in new_banned:로 코드를 수정하고 정답을 맞출 수 있었다.

 

나는 다음과 같이 중복을 처리하였는데

check = defaultdict(bool)
new_banned = []
for banned_num in banned:
	if not check[banned_num]:
		new_banned.append(banned_num)
		check[banned_num] = True

 

다른 사람들의 풀이를 보니 set으로 처리한 것 같았다.

확실히 내 코드보다는 set을 쓰는 것이 좀 더 효율적일 것 같다.

new_banned = set(banned)

 

---

[LeetCode] 2554. Maximum Number of Integers to Choose From a Range I