Driver Code
if __name__ == ' _ _ main _ _ ' : NEW_LINE INDENT k = 4 NEW_LINE root = newNode ( 1 ) NEW_LINE root . left = newNode ( 2 ) NEW_LINE root . right = newNode ( 3 ) NEW_LINE root . left . left = newNode ( 4 ) NEW_LINE root . left . right = newNode ( 5 ) NEW_LINE root . left . left . left = newNode ( 7 ) NEW_LINE root . right . right = newNode ( 6 ) NEW_LINE root . right . right . left = newNode ( 8 ) NEW_LINE print ( " Inorder ▁ Traversal ▁ of ▁ Original ▁ tree " ) NEW_LINE prInorder ( root ) NEW_LINE print ( ) NEW_LINE print ( " Inorder ▁ Traversal ▁ of ▁ Modified ▁ tree " ) NEW_LINE res = removeShortPathNodes ( root , k ) NEW_LINE prInorder ( res ) NEW_LINE DEDENT