Advent of Code 2016

--- Day 21: Scrambled Letters and Hash --- The computer system you're breaking into uses a weird scrambling function to store its passwords. It shouldn't be much trouble to create your own scrambled password so you can add it to the system; you just have to implement the scrambler. The scrambling function is a series of operations (the exact list is provided in your puzzle input). Starting with the password to be scrambled, apply each operation in succession to the string. The individual operations behave as follows: - swap position X with position Y means that the letters at indexes X and Y (counting from 0) should be swapped. - swap letter X with letter Y means that the letters X and Y should be swapped (regardless of where they appear in the string). - rotate left/right X steps means that the whole string should be rotated; for example, one right rotation would turn abcd into dabc. - rotate based on position of letter X means that the whole string should be rotated to the right based on the index of letter X (counting from 0) as determined before this instruction does any rotations. Once the index is determined, rotate the string to the right one time, plus a number of times equal to that index, plus one additional time if the index was at least 4. - reverse positions X through Y means that the span of letters at indexes X through Y (including the letters at X and Y) should be reversed in order. - move position X to position Y means that the letter which is at index X should be removed from the string, then inserted such that it ends up at index Y. For example, suppose you start with abcde and perform the following operations: - swap position 4 with position 0 swaps the first and last letters, producing the input for the next step, ebcda. - swap letter d with letter b swaps the positions of d and b: edcba. - reverse positions 0 through 4 causes the entire string to be reversed, producing abcde. - rotate left 1 step shifts all letters left one position, causing the first letter to wrap to the end of the string: bcdea. - move position 1 to position 4 removes the letter at position 1 (c), then inserts it at position 4 (the end of the string): bdeac. - move position 3 to position 0 removes the letter at position 3 (a), then inserts it at position 0 (the front of the string): abdec. - rotate based on position of letter b finds the index of letter b (1), then rotates the string right once plus a number of times equal to that index (2): ecabd. - rotate based on position of letter d finds the index of letter d (4), then rotates the string right once, plus a number of times equal to that index, plus an additional time because the index was at least 4, for a total of 6 right rotations: decab. After these steps, the resulting scrambled password is decab. Now, you just need to generate a new scrambled password and you can access the system. Given the list of scrambling operations in your puzzle input, what is the result of scrambling abcdefgh?

In [1]:
class Scramble:
    
    def __init__(self, password):
        self.password = list(password)        
        
    def swap_positions(self, x, y):
        """The letters at indexes X and Y (counting from 0) should be swapped."""
        self.password[x], self.password[y] = self.password[y], self.password[x]
    
    def swap_letters(self, x, y):
        """The letters X and Y should be swapped (regardless of where they appear in the string)."""
        x = self.password.index(x)
        y = self.password.index(y)
        self.swap_positions(x, y)
        
    def rotate_by_steps(self, how, steps):
        """The whole string should be rotated; for example, one right rotation would turn abcd into dabc."""
        pwd = list(self.password)
        for i in range(len(self.password)):
            if how == 'right':
                self.password[(i + steps) % len(self.password)] = pwd[i]
            else:
                self.password[(i - steps) % len(self.password)] = pwd[i]            

    def rotate_by_position(self, x):
        """The whole string should be rotated to the right based on the index 
        of letter X (counting from 0) as determined before this instruction 
        does any rotations. Once the index is  determined, rotate the string 
        to the right one time, plus a number of times equal to that index, plus 
        one additional time if the index was at least 4.
        """
        steps = self.password.index(x)
        steps += 2 if steps >= 4 else 1
        self.rotate_by_steps(how='right', steps=steps)
        
    def reverse(self, x, y):
        """The span of letters at indexes X through Y (including the 
        letters at X and Y) should be reversed in order.
        """
        sublist = self.password[x: y + 1]
        sublist.reverse()
        self.password = self.password[: x] + sublist + self.password[y + 1: ]
        
    def move(self, x, y):
        """The letter which is at index X should be removed from the string, 
        then inserted such that it ends up at index Y.
        """
        self.password.insert(y, self.password.pop(x))
    
    def get_scrambled_password(self):
        return ''.join(self.password)
    
    def run(self, data):
        for instruction in data:
            instruction = instruction.split()
            if instruction[0] == 'swap' and instruction[1] == 'position':
                x = int(instruction[2])
                y = int(instruction[-1])
                self.swap_positions(x, y)
            elif instruction[0] == 'swap' and instruction[1] == 'letter':
                x = instruction[2]
                y = instruction[-1]
                self.swap_letters(x, y)
            elif instruction[0] == 'rotate' and instruction[1] == 'based':
                x = instruction[-1]
                self.rotate_by_position(x)
            elif instruction[0] == 'rotate':
                how = instruction[1]
                steps = int(instruction[2])
                self.rotate_by_steps(how, steps)
            elif instruction[0] == 'reverse':
                x = int(instruction[2])
                y = int(instruction[-1])
                self.reverse(x, y)
            elif instruction[0] == 'move':
                x = int(instruction[2])
                y = int(instruction[-1])
                self.move(x, y)

In [2]:
data = [
    'swap position 4 with position 0',
    'swap letter d with letter b', 
    'reverse positions 0 through 4',
    'rotate left 1 step',
    'move position 1 to position 4',
    'move position 3 to position 0',
    'rotate based on position of letter b', 
    'rotate based on position of letter d'
]

scramble = Scramble('abcde')
scramble.run(data)
assert scramble.get_scrambled_password() == 'decab'

In [3]:
data = [line.strip() for line in open('data/day_21.txt', 'r')]

In [4]:
scramble = Scramble('abcdefgh')
scramble.run(data)
scramble.get_scrambled_password()


Out[4]:
'dgfaehcb'
--- Part Two --- You scrambled the password correctly, but you discover that you can't actually modify the password file on the system. You'll need to un-scramble one of the existing passwords by reversing the scrambling process. What is the un-scrambled version of the scrambled password fbgdceah?

In [5]:
import itertools
from tqdm import tqdm

In [6]:
def unscramble(password):
    res = []
    all_passwords = itertools.permutations(list(password), len(password))
    for pwd in tqdm(all_passwords):
        scramble = Scramble(''.join(pwd))
        scramble.run(data)
        if scramble.get_scrambled_password() == password:
            res.append(''.join(pwd))
    return res

In [7]:
data = [
    'swap position 4 with position 0',
    'swap letter d with letter b', 
    'reverse positions 0 through 4',
    'rotate left 1 step',
    'move position 1 to position 4',
    'move position 3 to position 0',
    'rotate based on position of letter b', 
    'rotate based on position of letter d'
]

assert 'abcde' in unscramble('decab')


0it [00:00, ?it/s]120it [00:00, 30399.01it/s]

In [8]:
data = [line.strip() for line in open('data/day_21.txt', 'r')]
unscramble('fbgdceah')


40320it [00:09, 4428.24it/s]00, 3143.64it/s]
Out[8]:
['fdhgacbe']