From ace71e8a98451b9fecb6e0a2d0efb6f04476f018 Mon Sep 17 00:00:00 2001 From: Michael Manring Date: Tue, 21 May 2024 03:32:18 +1000 Subject: pwc270 - improvement on ch-2 --- challenge-270/pokgopun/python/ch-2.py | 19 +++++++++---------- 1 file changed, 9 insertions(+), 10 deletions(-) (limited to 'challenge-270/pokgopun/python') 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 -- cgit