Commit 882ae26

authored

Merge pull request vJechsmayr#518 from anshul17024/insert

added solution for issue 0701

2 parents f606e7c + 49a6a74 commit 882ae26Copy full SHA for 882ae26

File tree

+34

-0

lines changed

+34

-0

lines changed

Lines changed: 34 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,34 @@`
	`1`	`+# Definition for a binary tree node.`
	`2`	`+# class TreeNode:`
	`3`	`+# def __init__(self, val=0, left=None, right=None):`
	`4`	`+# self.val = val`
	`5`	`+# self.left = left`
	`6`	`+# self.right = right`
	`7`	`+`
	`8`	`+def solve(r,k):`
	`9`	`+ x = TreeNode(k)`
	`10`	`+ ans = r`
	`11`	`+ if(not r):`
	`12`	`+ return x`
	`13`	`+ while(r and (r.left or r.right) ):`
	`14`	`+ if(r.val < k):`
	`15`	`+ if(r.right):`
	`16`	`+ r = r.right`
	`17`	`+ else:`
	`18`	`+ r.right = x`
	`19`	`+ return ans`
	`20`	`+ else:`
	`21`	`+ if(r.left):`
	`22`	`+ r = r.left`
	`23`	`+ else:`
	`24`	`+ r.left = x`
	`25`	`+ return ans`
	`26`	`+ if(r.val < k):`
	`27`	`+ r.right = x`
	`28`	`+ else:`
	`29`	`+ r.left = x`
	`30`	`+ return ans`
	`31`	`+`
	`32`	`+class Solution:`
	`33`	`+ def insertIntoBST(self, root: TreeNode, val: int) -> TreeNode:`
	`34`	`+ return solve(root,val)`

Comments

(0)