leetcode - insert into a binary search tree(kotlin)

silver·2021년 7월 6일
0

level - medium

자세한 설명과 내용은 아래 링크를 참고해주세요.
https://velog.io/@mdok1112/leetcode-insert-into-a-binary-search-treejava

위의 링크에서는 설명과 자바코드가 들어있으며
여기서는 코틀린으로 작성한 코드만 있습니다.

class Solution {
    fun insertIntoBST(root: TreeNode?, `val`: Int): TreeNode? {
        if(root == null) {
            return TreeNode(`val`)
        }

        var parent = root!!
        while(true) {
            if(parent.`val` < `val`) {
                if(parent.right == null) {
                    parent.right = TreeNode(`val`)
                    break
                }
                parent = parent.right!!
            } else if(parent.`val` > `val`) {
                if(parent.left == null) {
                    parent.left = TreeNode(`val`)
                    break
                }
                parent = parent.left!!
            }
        }

        return root
    }
}

0개의 댓글