1. 程式人生 > >Convert Sorted List to Binary Search Tree

Convert Sorted List to Binary Search Tree

Given a singly linked list where elements are sorted in ascending order, convert it to a height balanced BST.
想了好久想不出來,後來看了題目分類裡面說是DFS,但是沒有想出DFS的演算法來。後來想到了一個遞迴的方法,但是空間和時間都是O(n)。
後來網上找了一個空間是O(1)的時間是O(n)的演算法,是一種新的解題思路,用的是中遞迴。一般我解題都是用的頭遞迴或者尾遞迴,第一次
見識到了中遞迴,相當於把遞迴當成了一個迴圈體,用引用來作為變數,每個遞迴中修改,需要很強大的想象力,把整個遞迴樹在腦子裡想清楚。

空間和時間都為O(n):

    TreeNode *sortedListToBST(ListNode *head) 
	{
		vector<TreeNode*> treeNodes;
		while (head != NULL)
		{
			TreeNode *node = new TreeNode(head->val);
			treeNodes.push_back(node);
			head = head->next;
		}
		return genBST(0, treeNodes.size()-1, treeNodes);
    }
	
	TreeNode* genBST(int start, int end, vector<TreeNode*> &treeNodes)
	{
		if (start == end) return treeNodes[start];
		else if (start+1 == end)
		{
			treeNodes[start]->right = treeNodes[end];
			return treeNodes[start];
		} 

		int mid = (start+end)/2;
		TreeNode* root = treeNodes[mid];
		root->left = genBST(start, mid-1, treeNodes);
		root->right = genBST(mid+1, end, treeNodes);
		return root;
	}

空間為O(1)時間為O(n):
	TreeNode *sortedListToBST(ListNode *head)
	{
	    int len = 0;
        ListNode * node = head;
        while (node != NULL)
        {
            node = node->next;
            len++;
        }
        return buildTree(head, 0, len-1);
    }
    
    TreeNode *buildTree(ListNode *&node, int start, int end)
    {
        if (start > end) return NULL;
        int mid = start + (end - start)/2;
        TreeNode *left = buildTree(node, start, mid-1);
        TreeNode *root = new TreeNode(node->val);
        root->left = left;
        node = node->next;
        root->right = buildTree(node, mid+1, end);
        return root;
    }
解法引用:http://www.bwscitech.com/a/jishuzixun/javayuyan/2013/0930/15822.html