forked from ngiengkianyew/daily-coding-problem
-
Notifications
You must be signed in to change notification settings - Fork 0
/
problem_350.py
40 lines (31 loc) · 811 Bytes
/
problem_350.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
import sys
def get_sum_sq(target, squares):
if target == 0:
return 0
elif not squares:
return sys.maxsize
original_tgt = target
biggest_sq = squares.pop()
tally = 0
while target >= biggest_sq:
tally += 1
target -= biggest_sq
if tally:
return min(
tally + get_sum_sq(target, squares.copy()),
get_sum_sq(original_tgt, squares.copy())
)
else:
return get_sum_sq(original_tgt, squares.copy())
def get_min_squares(target):
num, sq = 1, 1
squares = list()
while sq <= target:
squares.append(sq)
num += 1
sq = num * num
return get_sum_sq(target, squares)
# Tests
assert get_min_squares(4) == 1
assert get_min_squares(17) == 2
assert get_min_squares(18) == 2