import sys
import os
sys.path.insert(1, os.path.abspath('../../'))
from python_tools.aoc_utils import *
import collections as C


def part_one(g):
    lan = set()
    for i in g:
        for key1 in g[i]:
            for key2 in g[i]:
                if key1 != key2:
                    if key2 in g[key1] and key1 in g[key2]:
                        pl = [i,key1, key2]
                        pl.sort()
                        lan.add(tuple(pl))

    p1 = 0
    for i in lan:
        for j in i:
            if j[0] == "t":
                p1 += 1
                break
    print("P1: ", p1)

def part_two(g):
    lan = set()
    for i in g:
        temp_lan = set([i])
        for key1 in g[i]:
            pc = True
            for j in temp_lan:
                if j not in g[key1]:
                    pc = False
            if pc:
                temp_lan.add(key1)
        x = list(temp_lan)
        x.sort()
        lan.add(','.join(x))

    p2 = 0
    max_val = []
    for i in lan:
        if len(i) > p2:
            p2 = len(i)
            max_val = i
    print("P2: ", max_val)

def main():
    contents = file2list("input.txt")

    g = C.defaultdict(set)

    for i in contents:
        x = i.split('-')
        g[x[0]].add(x[1])
        g[x[1]].add(x[0])
    part_one(g)
    part_two(g)

if __name__ == "__main__":
    main()