Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- >>> getNumbers(10)
- [100, 64, 36, 16, 4, 0, 4, 16, 36, 64, 100]
- >>> getNumbers(9)
- [81, 49, 25, 9, 1, 1, 9, 25, 49, 81]
- >>> getNumbers(8)
- [64, 36, 16, 4, 0, 4, 16, 36, 64]
- >>> getNumbers(0)
- [0]
- def getNumbers(num):
- x = []
- y = []
- if num % 2 == 0:
- x = [i**2 for i in range(0, num+2, 2)]
- y = [i**2 for i in range(0, num+2, 2)]
- z = sorted(x, reverse=True) + y
- if z.count(0) > 1:
- z.remove(0)
- return z
- elif num % 3 == 0:
- x = [i**2 for i in range(1, num+2, 2)]
- y = [i**2 for i in range(1, num+2, 2)]
- return sorted(x, reverse=True) + y
- elif num == 1:
- x.append(num)
- y.append(num)
- return sorted(x, reverse=True) + y
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement