■anond:20240816235943

高さ3の有向木で根からAという節点が生えAからBという葉が生える

class TreeNode:
    def __init__(self, name, attributes=None):
        self.name = name
        self.attributes = attributes or {}
        self.children = []

    def add_child(self, child_node):
        self.children.append(child_node)

    def display(self, level=0):
        indent = "    " * level
        print(f"{indent}{self.name} {self.attributes}")
        for child in self.children:
            child.display(level + 1)

    def has_dependency(self):
        # ルートノードや属性を持たないノードは依存関係を判定しない
        if not self.children or "start" not in self.attributes or "end" not in self.attributes:
            return False

        # Aノードのstartとendを取得
        start = self.attributes["start"]
        end = self.attributes["end"]

        # すべての子ノード（Bノード）に対して依存関係をチェック
        for child in self.children:
            if "position" in child.attributes:
                position = child.attributes["position"]
                if start <= position <= end:
                    print(f"{child.name} (position: {position}) is dependent on {self.name} (start: {start}, end: {end})")
                    return True
                else:
                    print(f"{child.name} (position: {position}) is NOT dependent on {self.name} (start: {start}, end: {end})")
        return False

# ノードの作成
root = TreeNode("Root")
a_node = TreeNode("A", {"start": 10, "end": 20})
b1_node = TreeNode("B1", {"position": 15})
b2_node = TreeNode("B2", {"position": 25})

# 木構造の構築
root.add_child(a_node)
a_node.add_child(b1_node)
a_node.add_child(b2_node)

# 木構造の表示
root.display()

# 依存関係のチェック
a_node.has_dependency()

こうかー

Permalink | 記事への反応(1) | 00:08

ツイートシェア