[LeetCode] 두 갈래 검색 트리 자르기(두 갈래 트리 자르기)

1536 단어

제목


두 갈래 검색 트리를 지정하고 최소 경계 L과 최대 경계 R을 지정합니다.두 갈래 검색 트리를 트림하여 모든 노드의 값을 [L, R](R>=L)로 만듭니다.너는 나무의 뿌리 노드를 바꿔야 할 수도 있으니, 결과는 잘린 두 갈래 검색 나무의 새로운 뿌리 노드로 돌아가야 한다.

생각


차례로 돌아가다.우선 현재 노드의value와 경계 간의 관계를 판단한다.비어 있으면 반복을 중지하고 비어 있는 상태로 돌아갑니다.노드의 값이 L보다 작으면 노드의 오른쪽 하위 트리를 되돌려주고, 노드의 값이 R보다 크면 오른쪽 노드가 귀속 함수를 호출하는 값을 되돌려줍니다.노드의 값이 L보다 작으면 왼쪽 노드 호출 귀속 함수의 값을 되돌려줍니다.마지막으로 노드의 값이 범위 내에 있으면 좌우 노드가 귀속 함수의 값을 호출하여 좌우 노드에 값을 부여하고 이 노드를 되돌려줍니다.

코드

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int L, int R) {
        if (root == NULL)   return NULL;
        if (root->val < L)  return trimBST(root->right, L, R);
        if (root->val > R)  return trimBST(root->left, L, R);
        root->left = trimBST(root->left, L, R);
        root->right = trimBST(root->right, L, R);
        return root;
    }
};

좋은 웹페이지 즐겨찾기