Advent of Code 2016

--- Day 6: Signals and Noise --- Something is jamming your communications with Santa. Fortunately, your signal is only partially jammed, and protocol in situations like this is to switch to a simple repetition code to get the message through. In this model, the same message is sent repeatedly. You've recorded the repeating message signal (your puzzle input), but the data seems quite corrupted - almost too badly to recover. Almost. All you need to do is figure out which character is most frequent for each position. For example, suppose you had recorded the following messages: eedadn drvtee eandsr raavrd atevrs tsrnev sdttsa rasrtv nssdts ntnada svetve tesnvt vntsnd vrdear dvrsen enarar The most common character in the first column is e; in the second, a; in the third, s, and so on. Combining these characters returns the error-corrected message, easter. Given the recording in your puzzle input, what is the error-corrected version of the message being sent?



In [1]:

    
from collections import Counter



In [2]:

    
def get_message(data):
    freqs = [Counter([line[i] for line in data]) for i in range(len(data[0]))]
    message = ''.join([freqs[i].most_common(1)[0][0] for i in range(len(freqs))])
    return message



In [3]:

    
# Test
data = [
    'eedadn',
    'drvtee',
    'eandsr',
    'raavrd',
    'atevrs',
    'tsrnev',
    'sdttsa',
    'rasrtv',
    'nssdts',
    'ntnada',
    'svetve',
    'tesnvt',
    'vntsnd',
    'vrdear',
    'dvrsen',
    'enarar'
]

get_message(data)









    Out[3]:





'easter'



In [4]:

    
data = [line.strip() for line in open('data/day_6.txt', 'r')]
get_message(data)









    Out[4]:





'mshjnduc'

--- Part Two --- Of course, that would be the message - if you hadn't agreed to use a modified repetition code instead. In this modified code, the sender instead transmits what looks like random data, but for each character, the character they actually want to send is slightly less likely than the others. Even after signal-jamming noise, you can look at the letter distributions in each column and choose the least common letter to reconstruct the original message. In the above example, the least common character in the first column is a; in the second, d, and so on. Repeating this process for the remaining characters produces the original message, advent. Given the recording in your puzzle input and this new decoding methodology, what is the original message that Santa is trying to send?



In [5]:

    
def get_message(data):
    freqs = [Counter([line[i] for line in data]) for i in range(len(data[0]))]
    message = ''.join([sorted(freqs[i].items(), key=lambda x: x[1])[0][0] for i in range(len(freqs))])
    return message



In [6]:

    
# Test
data = [
    'eedadn',
    'drvtee',
    'eandsr',
    'raavrd',
    'atevrs',
    'tsrnev',
    'sdttsa',
    'rasrtv',
    'nssdts',
    'ntnada',
    'svetve',
    'tesnvt',
    'vntsnd',
    'vrdear',
    'dvrsen',
    'enarar'
]

get_message(data)









    Out[6]:





'advent'



In [7]:

    
data = [line.strip() for line in open('data/day_6.txt', 'r')]
get_message(data)









    Out[7]:





'apfeeebz'