Пример #1
0
    def test_rightSideView_dfs1(self):
        node_root = TreeNode(1)
        node_root.left = TreeNode(2)
        node_root.left.right = TreeNode(5)

        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(2, right_side_view_list[1])
        self.assertEqual(5, 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)