-
Notifications
You must be signed in to change notification settings - Fork 3
/
Square_into_Squares_Protect_trees.py
64 lines (49 loc) · 2.28 KB
/
Square_into_Squares_Protect_trees.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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
# Kata link:
# https://www.codewars.com/kata/54eb33e5bc1a25440d000891/
# -------------------------------------
# Instructions:
'''
My little sister came back home from school with the following task: given a squared sheet of paper she has to cut it in pieces which, when assembled, give squares the sides of which form an increasing sequence of numbers. At the beginning it was lot of fun but little by little we were tired of seeing the pile of torn paper. So we decided to write a program that could help us and protects trees.
Task
Given a positive integral number n, return a strictly increasing sequence (list/array/string depending on the language) of numbers, so that the sum of the squares is equal to n².
If there are multiple solutions (and there will be), return as far as possible the result with the largest possible values:
Examples
decompose(11) must return [1,2,4,10]. Note that there are actually two ways to decompose 11², 11² = 121 = 1 + 4 + 16 + 100 = 1² + 2² + 4² + 10² but don't return [2,6,9], since 9 is smaller than 10.
For decompose(50) don't return [1, 1, 4, 9, 49] but [1, 3, 5, 8, 49] since [1, 1, 4, 9, 49] doesn't form a strictly increasing sequence.
Note
Neither [n] nor [1,1,1,…,1] are valid solutions. If no valid solution exists, return nil, null, Nothing, None (depending on the language) or "[]" (C) ,{} (C++), [] (Swift, Go).
The function "decompose" will take a positive integer n and return the decomposition of N = n² as:
[x1 ... xk] or
"x1 ... xk" or
Just [x1 ... xk] or
Some [x1 ... xk] or
{x1 ... xk} or
"[x1,x2, ... ,xk]"
depending on the language (see "Sample tests")
Note for Bash
decompose 50 returns "1,3,5,8,49"
decompose 4 returns "Nothing"
Hint
Very often xk will be n-1.
'''
# -------------------------------------
# Solution:
def decompose(n):
result = [n]
temp = 0
while result:
x = result.pop()
temp += x ** 2
for i in range(x - 1, 0, -1):
if temp - (i ** 2) >= 0:
temp -= i ** 2
result.append(i)
if temp == 0:
result.sort()
return result
return None
# -------------------------------------
# Basic Tests:
print(decompose(5), "=", [3,4])
print(decompose(8), "=", None)
print(decompose(50), "=", [1, 3, 5, 8, 49])