Commit 6f12319

author

Amogh Singhal

authored

Update bst_nodes_in_range.py

1 parent 8b84002 commit 6f12319Copy full SHA for 6f12319

File tree

-2

lines changed

-2

lines changed

Lines changed: 7 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -11,16 +11,21 @@ def __init__(self, data):`
`11`	`11`
`12`	`12`	`def nodesWithinRange(root, range):`
`13`	`13`	`low, high = range`
	`14`	`+ # this is the base case`
`14`	`15`	`if root is None:`
`15`	`16`	`return 0`
	`17`	`+ # optional: to improve efficiency`
	`18`	`+ elif root.data == high or root.data == low:`
	`19`	`+ return 1`
	`20`	`+ # if the current node lies in the range`
`16`	`21`	`elif root.data <= high and root.data >= low:`
`17`	`22`	`return (`
`18`	`23`	`1 + nodesWithinRange(root.left, range) + nodesWithinRange(root.right, range)`
`19`	`24`	`)`
`20`		`- elif root.data == high or root.data == low:`
`21`		`- return 1`
	`25`	`+ # if the current node lies in left subtree`
`22`	`26`	`elif root.data > high:`
`23`	`27`	`return nodesWithinRange(root.left, range)`
	`28`	`+ # if the current node lies in the right subtree`
`24`	`29`	`elif root.data < low:`
`25`	`30`	`return nodesWithinRange(root.right, range)`
`26`	`31`

Comments

(0)