lintCode(453)——将二叉树拆分成链表

题目描述：将一棵二叉树按照前序遍历拆解成为一个假链表。所谓的假链表是说，用二叉树的 right 指针，来表示链表中的 next 指针。

注意事项：不要忘记将左儿子标记为 null，否则你可能会得到空间溢出或是时间溢出。

样例
 

         1
        / \
       2   5
      / \   \
     3   4   6

The flattened tree should look like:

   1
    \
     2
      \
       3
        \
         4
          \
           5
            \
             6

挑战
不使用额外的空间耗费。
 

难度等级：容易

分析：本题采用递归的方法解决，关键是要知道由左子树转化的链表的头和尾，以及由右子树转化的链表的头和尾。头一定是二叉树的根节点，尾是右子树的尾（如果右子树不空）或者左子树的尾（如果右子树空，左子树不空）或者根（如果左右子树都是空）。

void flatten(TreeNode *root) {
// main function    

    if(root == NULL)
        return;
    ConvertToLink(root);
}


TreeNode* ConvertToLink(TreeNode* root) // utility function
{// return value is the tail of the linked list
    if(root==NULL)
        return NULL;

    TreeNode* leftLinkTail = ConvertToLink(root->left); // tail of left linked list
    TreeNode* rightLinkTail = ConvertToLink(root->right); // tail of left linked list
    
    if(leftLinkTail != NULL)
    {
        leftLinkTail->right = root->right;
        root->right = root->left;
    }
    
    root->left = NULL;

    if(rightLinkTail!= NULL)
        return rightLinkTail;
    else if(leftLinkTail != NULL)
        return leftLinkTail;
    else
        return root;
}

参考：http://blog.csdn.net/jiyanfeng1/article/details/41693439