Enable Javascript in your browser and then refresh this page, for a much enhanced experience.
First solution in Clear category for Can Balance by chunshar
from typing import Iterable
def can_balance(weights: Iterable) -> int:
torque = lambda x: sum((i + 1) * w for i, w in enumerate(x))
for f in range(1, len(weights) - 1):
if torque(reversed(weights[:f])) == torque(weights[f + 1:]):
return f
return 0 if len(weights) == 1 else -1
if __name__ == '__main__':
print("Example:")
print(can_balance([6, 1, 10, 5, 4]))
# These "asserts" are used for self-checking and not for an auto-testing
assert can_balance([6, 1, 10, 5, 4]) == 2
assert can_balance([10, 3, 3, 2, 1]) == 1
assert can_balance([7, 3, 4, 2, 9, 7, 4]) == -1
assert can_balance([42]) == 0
print("Coding complete? Click 'Check' to earn cool rewards!")
Dec. 19, 2018
