[LeetCode系列]有序链表转换为平衡BST的递归解法

给定有序链表(元素由小到大), 试问如何将其转换为一个平衡BST?

平衡BST: 任意节点的左右子树的深度差值不大于1.

主要思想是用递归. Trick是使用快慢指针来获取中间节点. 获得中间节点后, 将其设为此次递归的root, 随后删除此节点, 并将前一节点的next置NULL. 随后, 对中间节点的前后部分分别进行递归调用, 并将返回值作为其左右子树.

代码如下:

 /**
  * 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 == NULL) return NULL;
         if(head->next == NULL) return new TreeNode(head->val);
         ListNode *step1 = head;
         ListNode *step2 = head->next;
         while(step2->next != NULL && step2->next->next != NULL){
             step1 = step1->next;
             step2 = step2->next->next;
         }
         TreeNode *root  = new TreeNode(step1->next->val);
         ListNode *head2 = step1->next->next;
         delete step1->next;
         step1->next = NULL; // cut list into two parts
         root->left = sortedListToBST(head);
         root->right = sortedListToBST(head2);
         return root;
     }
 };