コード例 #1
0
    def test_rightSideView_dfs(self):
        node_root = TreeNode(1)
        node_root.left = TreeNode(2)
        node_root.right = TreeNode(3)
        node_root.left.right = TreeNode(5)
        node_root.right.right = TreeNode(4)

        right_side_view_list = self.solution.rightSideView_dfs(node_root)
        self.assertIsNotNone(right_side_view_list)
        self.assertEqual(3, len(right_side_view_list))
        self.assertEqual(1, right_side_view_list[0])
        self.assertEqual(3, right_side_view_list[1])
        self.assertEqual(4, right_side_view_list[2])
コード例 #2
0
    def test_lowestCommonAncestorBST(self):
        node_root = TreeNode(6)
        node_root.left = TreeNode(2)
        node_root.left.left = TreeNode(0)
        node_root.left.right = TreeNode(4)
        node_root.left.right.left = TreeNode(3)
        node_root.left.right.right = TreeNode(5)
        node_root.right = TreeNode(8)
        node_root.right.left = TreeNode(7)
        node_root.right.right = TreeNode(9)

        lca = self.solution.lowestCommonAncestorBST(node_root, node_root.left, node_root.right)
        self.assertIsNotNone(lca)
        self.assertEqual(node_root, lca)