[LeetCode] Assign Cookies

문제 링크 >> https://leetcode.com/problems/assign-cookies/

 

 

📋 문제

Assume you are an awesome parent and want to give your children some cookies.

But, you should give each child at most one cookie.

 

Each child i has a greed factor g[i], which is the minimum size of a cookie that the child will be content with; and each cookie j has a size s[j]. If s[j] >= g[i], we can assign the cookie j to the child i, and the child i will be content. Your goal is to maximize the number of your content children and output the maximum number.

 

 

👉 입출력

Example 1:

Input: g = [1,2,3], s = [1,1]
Output: 1
Explanation: You have 3 children and 2 cookies. The greed factors of 3 children are 1, 2, 3. 
And even though you have 2 cookies, since their size is both 1, you could only make the child whose greed factor is 1 content.
You need to output 1.

Example 2:

Input: g = [1,2], s = [1,2,3]
Output: 2
Explanation: You have 2 children and 3 cookies. The greed factors of 2 children are 1, 2. 
You have 3 cookies and their sizes are big enough to gratify all of the children, 
You need to output 2.

 

Constraints:

• 1 <= g.length <= 3 * 104
• 0 <= s.length <= 3 * 104
• 1 <= g[i], s[j] <= 231 - 1

 

 

📝 풀이

class Solution:
    def findContentChildren(self, g: List[int], s: List[int]) -> int:
        g.sort()
        s.sort()
        
        count = 0
        i = 0
        for j in range(len(s)):
            if i < len(g) and s[j] >= g[i]:
                count += 1
                i += 1
        return count

 

먼저 아이가 필요한 쿠키의 양에 최대한 맞추기 위해 인자로 주어진 두 리스트 g와 s를 오름차순으로 정렬했다.

그리고 가지고 있는 쿠키 리스트인 s를 순회하며 g와 비교해 나간다.

만약 g[i]가 g의 범위 안에 있고 쿠키를 줄 수 있는 경우라면 count에 1을 더하고 다음 아이가 원하는 쿠키의 양을 살핀다.

이 경우가 바로 쿠키를 아이에게 줄 수 있는 경우이다.

이 경우에 해당하지 않는다면 쿠키를 줄 수 없으므로 넘어간다.