diff options
Diffstat (limited to 'challenge-270/pokgopun/python/ch-2.py')
| -rw-r--r-- | challenge-270/pokgopun/python/ch-2.py | 19 |
1 files changed, 9 insertions, 10 deletions
diff --git a/challenge-270/pokgopun/python/ch-2.py b/challenge-270/pokgopun/python/ch-2.py index ac6e3bbcb0..ad5035105a 100644 --- a/challenge-270/pokgopun/python/ch-2.py +++ b/challenge-270/pokgopun/python/ch-2.py @@ -76,20 +76,19 @@ def distElem(ints: list, x: int, y: int): mx = ints[-1] c = 0 while True: - ints = ints[:ints.index(mx)] - l = len(ints) + l = ints.index(mx) if l == 0: break + d = mx - ints[l-1] if l == 1 or 2*x < y: - while ints[-1] < mx: - for i in range(l): - ints[i] += 1 - c += x + for i in range(l): + ints[i] += d + c += x * d else: - while ints[-1] < mx: - for i in range(-2,0): - ints[i] += 1 - c += y + p = l // 2 + for i in range(2*p): + ints[l-1-i] += d + c += y * p * d return c import unittest |