Cleanup Python code
Signed-off-by: AKU <tom@tdpain.net>
This commit is contained in:
parent
157f959bd8
commit
6e8a4c7364
1 changed files with 11 additions and 12 deletions
|
@ -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)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue