part_b.py

#!/usr/bin/env python3
import utils

from year_2018.day_08 import part_a


class Challenge(utils.BaseChallenge):
    def solve(self, _input, debug=False):
        """
        >>> Challenge().default_solve()
        20839
        """
        return NodeExtended.from_nodes_text(_input).get_value()


class NodeExtended(part_a.Node):
    def get_value(self):
        """
        >>> NodeExtended\\
        ...     .from_nodes_text("2 3 0 3 10 11 12 1 1 0 1 99 2 1 1 2")\\
        ...     .get_value()
        66
        >>> NodeExtended((), ()).get_value()
        0
        >>> NodeExtended((), (1, 2, 3)).get_value()
        6
        >>> NodeExtended((NodeExtended((), (1, 2, 3)),), (1,)).get_value()
        6
        >>> NodeExtended((NodeExtended((), (1, 2, 3)),), (1, 1)).get_value()
        12
        >>> NodeExtended((NodeExtended((), (1, 2, 3)),), (2, 1, 2)).get_value()
        6
        """
        if self.children:
            children_values = {
                index: (
                    self.children[index - 1].get_value()
                    if index - 1 < len(self.children) else
                    0
                )
                for index in set(self.metadata)
            }
            return sum(
                children_values[index]
                for index in self.metadata
            )
        else:
            return self.metadata_sum()


Challenge.main()
challenge = Challenge()