seongbeom_park/google-foobar-challenge
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

exclude if gear size is less than 1

Browse Source
This commit is contained in:
Seongbeom Park 2022-05-09 08:26:40 +09:00
parent 49c892ef9b
commit f287551e1c
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
extra/gearing-up-for-destruction/solution.py
View File

@ -3,7 +3,7 @@ def solution(pegs):
if len(pegs) % 2 == 0:
s = 2 * odd - 2 * even - pegs[-1] + pegs[0]
if s <= 0:
if 2*s < 3:
return [-1, -1]
if s % 3 == 0:
return [2*s/3, 1]
@ -11,7 +11,7 @@ def solution(pegs):
return [2*s, 3]
else:
s = 2 * odd - 2 * even + pegs[-1] + pegs[0]
if s <= 0:
if 2*s < 1:
return [-1, -1]
return [2*s, 1]
@ -28,6 +28,8 @@ def sum_odd_even(pegs):
tests = [
([4, 30, 50], [12, 1]),
([4, 17, 50], [-1, -1]),
([1, 2], [-1, -1]),
([1, 2, 3], [-1, -1]),
]
for i, o in tests: