Cleanup Python code

Signed-off-by: AKU <tom@tdpain.net>
This commit is contained in:
akp 2021-12-01 07:43:08 +00:00
parent 157f959bd8
commit 6e8a4c7364
No known key found for this signature in database
GPG key ID: AA5726202C8879B7

View file

@ -6,23 +6,22 @@ def parse(instr: str) -> List[int]:
return [int(x) for x in instr.splitlines() if x != ""]
class Challenge(BaseChallenge):
def count_increases(data: List[int]) -> int:
c = 0
for i in range(1, len(data)):
if data[i] > data[i - 1]:
c += 1
return c
class Challenge(BaseChallenge):
@staticmethod
def one(instr: str) -> int:
data = parse(instr)
c = 0
for i in range(1, len(data)):
if data[i] > data[i-1]:
c += 1
return c
return count_increases(data)
@staticmethod
def two(instr: str) -> int:
data = parse(instr)
c = 0
sums = [sum(data[i:i+3]) for i in range(len(data)-2)]
for i in range(1, len(sums)):
if sums[i] > sums[i-1]:
c += 1
return c
sums = [sum(data[i : i + 3]) for i in range(len(data) - 2)]
return count_increases(sums)