convert-sorted-list-to-binary-search-tree

1308 단어 알고리즘 문제

제목 설명


Given a singly linked list where elements are sorted in ascending order, convert it to a height balanced BST.
사고방식: 빠른 바늘을 통해 중간 결점을 찾아 중간 결점을 두 갈래 검색 트리에 삽입한 다음에 중간에서 질서정연한 단사슬표를 두 부분으로 뜯어내고 앞부분은 왼쪽 나무, 뒷부분은 오른쪽 나무로 분류한다.
/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode *sortedListToBST(ListNode *head) {
        if(!head){
            return nullptr;
        }
        if(!head->next){
            return new TreeNode(head->val);
        }
        ListNode* fast = head;
        ListNode* slow = head;
        ListNode* pre = head;
        while(fast && fast->next){
            fast = fast->next->next;
            pre = slow;
            slow = slow->next;
        }
        TreeNode* root = new TreeNode(slow->val);
        pre->next = nullptr;
        root->left = sortedListToBST(head);
        root->right = sortedListToBST(slow->next);
        return root;
    }
};

좋은 웹페이지 즐겨찾기