def mid(values: list[str]):
    return values[len(values)//2]

rules,updates = {}, []
with open('sample.data') as data:
    for line in data:
        if '|' in line:
            left, right  = [int(i) for i in line.split('|')]
            if left in rules:
                rules[left].append(right)
            else:
                rules[left] = [right]
        if ',' in line:
            updates.append([int(i) for i in line.split(',')])

def sort_updates(rules: dict[int, list[int]], updates: list) -> [list[int], list[int]]:
    correct, incorrect = [], []
    for update in updates:
        for idx, val in enumerate(update):
            fail = False
            for after in update[idx:]:
                if after == val:
                    continue
                try:
                    if after not in rules[val]:
                        fail = True
                        incorrect.append(update)
                        break
                except KeyError:
                    fail = True
                    incorrect.append(update)
                    break
            if fail:
                break
        else:
            # 75,47,61,53,29
            # 97,61,53,29,13
            # 75,29,13
            print('good', update)
            correct.append(mid(update))
    return correct, incorrect

correct, incorrect = sort_updates(rules, updates)
print('COrrect', sum(correct))
print('incorrect', *incorrect)