--- Day 13: Knights of the Dinner Table --- In years past, the holiday feast with your family hasn't gone so well. Not everyone gets along! This year, you resolve, will be different. You're going to find the optimal seating arrangement and avoid all those awkward conversations. You start by writing up a list of everyone invited and the amount their happiness would increase or decrease if they were to find themselves sitting next to each other person. You have a circular table that will be just big enough to fit everyone comfortably, and so each person will have exactly two neighbors. For example, suppose you have only four attendees planned, and you calculate their potential happiness as follows: Alice would gain 54 happiness units by sitting next to Bob. Alice would lose 79 happiness units by sitting next to Carol. Alice would lose 2 happiness units by sitting next to David. Bob would gain 83 happiness units by sitting next to Alice. Bob would lose 7 happiness units by sitting next to Carol. Bob would lose 63 happiness units by sitting next to David. Carol would lose 62 happiness units by sitting next to Alice. Carol would gain 60 happiness units by sitting next to Bob. Carol would gain 55 happiness units by sitting next to David. David would gain 46 happiness units by sitting next to Alice. David would lose 7 happiness units by sitting next to Bob. David would gain 41 happiness units by sitting next to Carol. Then, if you seat Alice next to David, Alice would lose 2 happiness units (because David talks so much), but David would gain 46 happiness units (because Alice is such a good listener), for a total change of 44. If you continue around the table, you could then seat Bob next to Alice (Bob gains 83, Alice gains 54). Finally, seat Carol, who sits next to Bob (Carol gains 60, Bob loses 7) and David (Carol gains 55, David gains 41). The arrangement looks like this: +41 +46 +55 David -2 Carol Alice +60 Bob +54 -7 +83 After trying every other seating arrangement in this hypothetical scenario, you find that this one is the most optimal, with a total change in happiness of 330. What is the total change in happiness for the optimal seating arrangement of the actual guest list?



In [50]:

    
import re
from collections import defaultdict
from itertools import permutations

_PARSE = re.compile("([A-Za-z]+) would (gain|lose) ([0-9]+) happiness units by sitting next to ([A-Za-z]+).")

def count_happines(units, order):
    p = order[0]    
    v_l = units[p][order[-1]]
    for c in order[1:]:
        v_l += int(units[p][c])
        p = c
        
    p = order[-1]    
    v_r = units[p][order[0]]
    for c in order[::-1][1:]:
        v_r += int(units[p][c])
        p = c

    return v_r + v_l

def happiness(data, addme=False):    
    units = defaultdict(dict)    
    for d in data:
        name1, gain_loss, points, name2 = _PARSE.match(d).groups()
        points = int(points) if gain_loss == 'gain' else -int(points)
        units[name1][name2] = points
        
    if addme:
        for name in list(units.keys()):
            units['me'][name] = 0
            units[name]['me'] = 0
        
    names = list(units.keys())
    
    max_perm = tuple(names)
    max_value = count_happines(units, max_perm)
    
    for perm in permutations(units.keys()):
        value = count_happines(units, perm)
        if value > max_value:
            max_perm = perm
            max_value = value
    
    return max_value, max_perm



In [51]:

    
# Test example
example = [
    "Alice would gain 54 happiness units by sitting next to Bob.",
    "Alice would lose 79 happiness units by sitting next to Carol.",
    "Alice would lose 2 happiness units by sitting next to David.",
    "Bob would gain 83 happiness units by sitting next to Alice.",
    "Bob would lose 7 happiness units by sitting next to Carol.",
    "Bob would lose 63 happiness units by sitting next to David.",
    "Carol would lose 62 happiness units by sitting next to Alice.",
    "Carol would gain 60 happiness units by sitting next to Bob.",
    "Carol would gain 55 happiness units by sitting next to David.",
    "David would gain 46 happiness units by sitting next to Alice.",
    "David would lose 7 happiness units by sitting next to Bob.",
    "David would gain 41 happiness units by sitting next to Carol."
]
happiness(example)









    Out[51]:





(330, ('Bob', 'Carol', 'David', 'Alice'))



In [52]:

    
# Test input
with open('day13/input.txt', 'rt') as fd:
    print(happiness(fd))









    



(618, ('Bob', 'Carol', 'David', 'George', 'Mallory', 'Frank', 'Alice', 'Eric'))

--- Part Two --- In all the commotion, you realize that you forgot to seat yourself. At this point, you're pretty apathetic toward the whole thing, and your happiness wouldn't really go up or down regardless of who you sit next to. You assume everyone else would be just as ambivalent about sitting next to you, too. So, add yourself to the list, and give all happiness relationships that involve you a score of 0. What is the total change in happiness for the optimal seating arrangement that actually includes yourself?



In [53]:

    
with open('day13/input.txt', 'rt') as fd:
    print(happiness(fd, addme=True))









    



(601, ('Bob', 'Carol', 'David', 'George', 'Mallory', 'Frank', 'Alice', 'Eric', 'me'))



In [ ]: