■anond:20240817001425

で・・・できたけど木の組み立てがしんどすぎるー

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:
            return False

        for child in self.children:
            # 子ノードがBusinessHourかScheduleかをチェック
            if "start_at" in child.attributes and "end_at" in child.attributes:
                child_start = child.attributes["start_at"]
                child_end = child.attributes["end_at"]

                # 現在のノードがBusinessHourで、子がScheduleの場合
                if "start_at" in self.attributes and "end_at" in self.attributes:
                    self_start = self.attributes["start_at"]
                    self_end = self.attributes["end_at"]

                    if self_start <= child_start and self_end >= child_end:
                        print(f"{child.name} (start_at: {child_start}, end_at: {child_end}, room_id: {child.attributes['room_id']}) is dependent on {self.name} (start_at: {self_start}, end_at: {self_end})")
                    else:
                        print(f"{child.name} (start_at: {child_start}, end_at: {child_end}, room_id: {child.attributes['room_id']}) is NOT dependent on {self.name} (start_at: {self_start}, end_at: {self_end})")

                # 現在のノードがRoomで、子がScheduleの場合
                elif self.name.startswith("Room"):
                    print(f"{child.name} (start_at: {child_start}, end_at: {child_end}, room_id: {child.attributes['room_id']}) is dependent on Room {self.name[-1]}")
                else:
                    child.has_dependency()

            # 子ノードが属性を持たない場合、再帰的に依存関係をチェック
            else:
                child.has_dependency()

# ノードの作成
root = TreeNode("Root")
office_node = TreeNode("Office")

# Roomノードの作成
room1_node = TreeNode("Room1")
room2_node = TreeNode("Room2")

# BusinessHourノードの作成
business_hour1_node = TreeNode("BusinessHour1", {"start_at": 9, "end_at": 12})
business_hour2_node = TreeNode("BusinessHour2", {"start_at": 13, "end_at": 17})

# Scheduleノードの作成
schedule1_node = TreeNode("Schedule1", {"start_at": 10, "end_at": 11, "room_id": 1})
schedule2_node = TreeNode("Schedule2", {"start_at": 14, "end_at": 15, "room_id": 1})
schedule3_node = TreeNode("Schedule3", {"start_at": 10, "end_at": 11, "room_id": 2})
schedule4_node = TreeNode("Schedule4", {"start_at": 14, "end_at": 15, "room_id": 2})

# 木構造の構築
root.add_child(office_node)
office_node.add_child(room1_node)
office_node.add_child(room2_node)
office_node.add_child(business_hour1_node)
office_node.add_child(business_hour2_node)

# Room1にSchedule1, Schedule2を追加
room1_node.add_child(schedule1_node)
room1_node.add_child(schedule2_node)

# Room2にSchedule3, Schedule4を追加
room2_node.add_child(schedule3_node)
room2_node.add_child(schedule4_node)

# BusinessHour1にSchedule1, Schedule3を追加
business_hour1_node.add_child(schedule1_node)
business_hour1_node.add_child(schedule3_node)

# BusinessHour2にSchedule2, Schedule4を追加
business_hour2_node.add_child(schedule2_node)
business_hour2_node.add_child(schedule4_node)

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

# 依存関係のチェック
office_node.has_dependency()
room1_node.has_dependency()
room2_node.has_dependency()
business_hour1_node.has_dependency()
business_hour2_node.has_dependency()

Root {}
    Office {}
        Room1 {}
            Schedule1 {'start_at': 10, 'end_at': 11, 'room_id': 1}
            Schedule2 {'start_at': 14, 'end_at': 15, 'room_id': 1}
        Room2 {}
            Schedule3 {'start_at': 10, 'end_at': 11, 'room_id': 2}
            Schedule4 {'start_at': 14, 'end_at': 15, 'room_id': 2}
        BusinessHour1 {'start_at': 9, 'end_at': 12}
            Schedule1 {'start_at': 10, 'end_at': 11, 'room_id': 1}
            Schedule3 {'start_at': 10, 'end_at': 11, 'room_id': 2}
        BusinessHour2 {'start_at': 13, 'end_at': 17}
            Schedule2 {'start_at': 14, 'end_at': 15, 'room_id': 1}
            Schedule4 {'start_at': 14, 'end_at': 15, 'room_id': 2}
Schedule1 (start_at: 10, end_at: 11, room_id: 1) is dependent on Room 1
Schedule2 (start_at: 14, end_at: 15, room_id: 1) is dependent on Room 1
Schedule3 (start_at: 10, end_at: 11, room_id: 2) is dependent on Room 2
Schedule4 (start_at: 14, end_at: 15, room_id: 2) is dependent on Room 2
Schedule1 (start_at: 10, end_at: 11, room_id: 1) is dependent on BusinessHour1 (start_at: 9, end_at: 12)
Schedule3 (start_at: 10, end_at: 11, room_id: 2) is dependent on BusinessHour1 (start_at: 9, end_at: 12)
Schedule2 (start_at: 14, end_at: 15, room_id: 1) is dependent on BusinessHour2 (start_at: 13, end_at: 17)
Schedule4 (start_at: 14, end_at: 15, room_id: 2) is dependent on BusinessHour2 (start_at: 13, end_at: 17)
Schedule1 (start_at: 10, end_at: 11, room_id: 1) is dependent on Room 1
Schedule2 (start_at: 14, end_at: 15, room_id: 1) is dependent on Room 1
Schedule3 (start_at: 10, end_at: 11, room_id: 2) is dependent on Room 2
Schedule4 (start_at: 14, end_at: 15, room_id: 2) is dependent on Room 2
Schedule1 (start_at: 10, end_at: 11, room_id: 1) is dependent on BusinessHour1 (start_at: 9, end_at: 12)
Schedule3 (start_at: 10, end_at: 11, room_id: 2) is dependent on BusinessHour1 (start_at: 9, end_at: 12)
Schedule2 (start_at: 14, end_at: 15, room_id: 1) is dependent on BusinessHour2 (start_at: 13, end_at: 17)
Schedule4 (start_at: 14, end_at: 15, room_id: 2) is dependent on BusinessHour2 (start_at: 13, end_at: 17)

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

ツイートシェア