Fair Distribution of Cookies LeetCode solution in Python

1
class Solution:
2
    def distributeCookies(self, cookies: List[int], k: int) -> int:
3
        ans = float('inf')
4
        fair = [0]*k
5
        def rec(i):
6
            nonlocal ans,fair
7
            if i == len(cookies):
8
                ans = min(ans,max(fair))
9
                return
10
			# Bounding condition to stop a branch if unfairness already exceeds current optimal soltution
11
			if ans <= max(fair):
12
                return
13
            for j in range(k):
14
                fair[j] += cookies[i]
15
                rec(i+1)
16
                fair[j] -= cookies[i]
17
        rec(0)
18
        return ans