AOC 2018 - Day 4 - Python

aoc2018_day4.py

# Prepare and sort data

data = [line.strip() for line in open('day4.txt')]
data.sort(key=lambda x: x.split(']')[0])

# Part 1

sleepingGuards = {}
currentGuard = None
asleepFrom = None

for line in data:
    date, time, keyword, guard = line.split(' ')[:4]
    if keyword == 'Guard':
        currentGuard = int(guard[1:])
    elif keyword == 'falls':
        asleepFrom = int(time[3:5])
    elif keyword == 'wakes':
        if currentGuard not in sleepingGuards:
            sleepingGuards[currentGuard] = {}
        for i in range(asleepFrom, int(time[3:5])):
            if i not in sleepingGuards[currentGuard]:
                sleepingGuards[currentGuard][i] = 0
            sleepingGuards[currentGuard][i] += 1
        asleepFrom = None

asleepTimes = {guard: sum(minutes.values()) for guard, minutes in sleepingGuards.items()}
asleepMost = {guard: max(minutes, key=minutes.get) for guard, minutes in sleepingGuards.items()}
mostAsleepGuard = max(asleepTimes, key=asleepTimes.get)
print(int(mostAsleepGuard * asleepMost[mostAsleepGuard]))

# Part 2

asleepMostOnSameMinute = {guard: max(minutes.values()) for guard, minutes in sleepingGuards.items()}
mostAsleepGuard = max(asleepMostOnSameMinute, key=asleepMostOnSameMinute.get)
mostAsleepMinute = {guard: max(minutes, key=minutes.get) for guard, minutes in sleepingGuards.items()}
print(int(mostAsleepGuard * mostAsleepMinute[mostAsleepGuard]))