19 lines
733 B
Python
19 lines
733 B
Python
def index_power(array: list, n: int) -> int:
|
|
"""
|
|
Find Nth power of the element with index N.
|
|
"""
|
|
return (array[n] ** n) if n < len(array) else -1
|
|
|
|
|
|
if __name__ == '__main__':
|
|
print('Example:')
|
|
print(index_power([1, 2, 3, 4], 2))
|
|
|
|
# These "asserts" using only for self-checking and not necessary for auto-testing
|
|
assert index_power([1, 2, 3, 4], 2) == 9, "Square"
|
|
assert index_power([1, 3, 10, 100], 3) == 1000000, "Cube"
|
|
assert index_power([0, 1], 0) == 1, "Zero power"
|
|
assert index_power([1, 2], 3) == -1, "IndexError"
|
|
assert index_power([1, 1, 1, 1, 1, 1, 1, 1, 1, 1], 9) == 1, "IndexError"
|
|
print("Coding complete? Click 'Check' to review your tests and earn cool rewards!")
|