📜  将给定的二叉树转换为双向链表 |设置 1

📅  最后修改于: 2022-05-13 01:57:17.360000             🧑  作者: Mango

将给定的二叉树转换为双向链表 |设置 1

Given a Binary Tree (Bt), convert it to a Doubly Linked List(DLL). The left and right pointers in nodes are to be used as previous and next pointers respectively in converted DLL. The order of nodes in DLL must be the same as in Inorder for the given Binary Tree. The first node of Inorder traversal (leftmost node in BT) must be the head node of the DLL. 

树到列表

我在一次采访中遇到了这个问题。在这篇文章中讨论了一个类似的问题。

这里的问题更简单,因为我们不需要创建一个循环 DLL,而是一个简单的 DLL。其解决方案背后的想法非常简单明了。

  1. 如果左子树存在,则处理左子树
    1. 递归地将左子树转换为 DLL。
    2. 然后在左子树中找到根的有序前驱(有序前驱是左子树中最右边的节点)。
    3. 将有序前驱作为前一个根,将根作为下一个有序前驱。
  2.  如果存在右子树,则处理右子树(以下3个步骤与左子树类似)。
    1. 递归地将右子树转换为 DLL。
    2. 然后在右子树中找到根的中序后继(按顺序后继是右子树中最左边的节点)。
    3. 将中序后继者作为下一个根,将根作为前一个中序后继。
  3. 找到最左边的节点并将其返回(最左边的节点始终是转换后的 DLL 的头部)。

下面是上述算法的源代码。

C++
// A C++ program for in-place
// conversion of Binary Tree to DLL
#include 
using namespace std;
 
/* A binary tree node has data,
and left and right pointers */
class node {
public:
    int data;
    node* left;
    node* right;
};
 
/* This is the core function to convert
Tree to list. This function follows
steps 1 and 2 of the above algorithm */
node* bintree2listUtil(node* root)
{
    // Base case
    if (root == NULL)
        return root;
 
    // Convert the left subtree and link to root
    if (root->left != NULL) {
        // Convert the left subtree
        node* left = bintree2listUtil(root->left);
 
        // Find inorder predecessor. After this loop, left
        // will point to the inorder predecessor
        for (; left->right != NULL; left = left->right)
            ;
 
        // Make root as next of the predecessor
        left->right = root;
 
        // Make predecessor as previous of root
        root->left = left;
    }
 
    // Convert the right subtree and link to root
    if (root->right != NULL) {
        // Convert the right subtree
        node* right = bintree2listUtil(root->right);
 
        // Find inorder successor. After this loop, right
        // will point to the inorder successor
        for (; right->left != NULL; right = right->left)
            ;
 
        // Make root as previous of successor
        right->left = root;
 
        // Make successor as next of root
        root->right = right;
    }
 
    return root;
}
 
// The main function that first calls
// bintree2listUtil(), then follows step 3
// of the above algorithm
node* bintree2list(node* root)
{
    // Base case
    if (root == NULL)
        return root;
 
    // Convert to DLL using bintree2listUtil()
    root = bintree2listUtil(root);
 
    // bintree2listUtil() returns root node of the converted
    // DLL. We need pointer to the leftmost node which is
    // head of the constructed DLL, so move to the leftmost
    // node
    while (root->left != NULL)
        root = root->left;
 
    return (root);
}
 
/* Helper function that allocates a new node with the
given data and NULL left and right pointers. */
node* newNode(int data)
{
    node* new_node = new node();
    new_node->data = data;
    new_node->left = new_node->right = NULL;
    return (new_node);
}
 
/* Function to print nodes in a given doubly linked list */
void printList(node* node)
{
    while (node != NULL) {
        cout << node->data << " ";
        node = node->right;
    }
}
 
/* Driver code*/
int main()
{
    // Let us create the tree shown in above diagram
    node* root = newNode(10);
    root->left = newNode(12);
    root->right = newNode(15);
    root->left->left = newNode(25);
    root->left->right = newNode(30);
    root->right->left = newNode(36);
 
    // Convert to DLL
    node* head = bintree2list(root);
 
    // Print the converted list
    printList(head);
 
    return 0;
}
 
// This code is contributed by rathbhupendra


C
// A C program for in-place conversion of Binary Tree to DLL
#include 
 
/* A binary tree node has data, and left and right pointers */
struct node
{
    int data;
    node* left;
    node* right;
};
 
/* This is the core function to convert Tree to list. This function follows
  steps 1 and 2 of the above algorithm */
node* bintree2listUtil(node* root)
{
    // Base case
    if (root == NULL)
        return root;
 
    // Convert the left subtree and link to root
    if (root->left != NULL)
    {
        // Convert the left subtree
        node* left = bintree2listUtil(root->left);
 
        // Find inorder predecessor. After this loop, left
        // will point to the inorder predecessor
        for (; left->right!=NULL; left=left->right);
 
        // Make root as next of the predecessor
        left->right = root;
 
        // Make predecessor as previous of root
        root->left = left;
    }
 
    // Convert the right subtree and link to root
    if (root->right!=NULL)
    {
        // Convert the right subtree
        node* right = bintree2listUtil(root->right);
 
        // Find inorder successor. After this loop, right
        // will point to the inorder successor
        for (; right->left!=NULL; right = right->left);
 
        // Make root as previous of successor
        right->left = root;
 
        // Make successor as next of root
        root->right = right;
    }
 
    return root;
}
 
// The main function that first calls bintree2listUtil(), then follows step 3
//  of the above algorithm
node* bintree2list(node *root)
{
    // Base case
    if (root == NULL)
        return root;
 
    // Convert to DLL using bintree2listUtil()
    root = bintree2listUtil(root);
 
    // bintree2listUtil() returns root node of the converted
    // DLL.  We need pointer to the leftmost node which is
    // head of the constructed DLL, so move to the leftmost node
    while (root->left != NULL)
        root = root->left;
 
    return (root);
}
 
/* Helper function that allocates a new node with the
   given data and NULL left and right pointers. */
node* newNode(int data)
{
    node* new_node = new node;
    new_node->data = data;
    new_node->left = new_node->right = NULL;
    return (new_node);
}
 
/* Function to print nodes in a given doubly linked list */
void printList(node *node)
{
    while (node!=NULL)
    {
        printf("%d ", node->data);
        node = node->right;
    }
}
 
/* Driver program to test above functions*/
int main()
{
    // Let us create the tree shown in above diagram
    node *root        = newNode(10);
    root->left        = newNode(12);
    root->right       = newNode(15);
    root->left->left  = newNode(25);
    root->left->right = newNode(30);
    root->right->left = newNode(36);
 
    // Convert to DLL
    node *head = bintree2list(root);
 
    // Print the converted list
    printList(head);
 
    return 0;
}


Java
// Java program to convert binary tree to double linked list
  
/* A binary tree node has data, and left and right pointers */
class Node
{
    int data;
    Node left, right;
  
    Node(int item)
    {
        data = item;
        left = right = null;
    }
}
  
class BinaryTree
{
    Node root;
    /* This is the core function to convert Tree to list. This function
       follows steps 1 and 2 of the above algorithm */
  
    Node bintree2listUtil(Node node)
    {
        // Base case
        if (node == null)
            return node;
  
        // Convert the left subtree and link to root
        if (node.left != null)
        {
            // Convert the left subtree
            Node left = bintree2listUtil(node.left);
  
            // Find inorder predecessor. After this loop, left
            // will point to the inorder predecessor
            for (; left.right != null; left = left.right);
  
            // Make root as next of the predecessor
            left.right = node;
  
            // Make predecessor as previous of root
            node.left = left;
        }
  
        // Convert the right subtree and link to root
        if (node.right != null)
        {
            // Convert the right subtree
            Node right = bintree2listUtil(node.right);
  
            // Find inorder successor. After this loop, right
            // will point to the inorder successor
            for (; right.left != null; right = right.left);
  
            // Make root as previous of successor
            right.left = node;
  
            // Make successor as next of root
            node.right = right;
        }
  
        return node;
    }
  
    // The main function that first calls bintree2listUtil(), then follows
    // step 3 of the above algorithm
      
    Node bintree2list(Node node)
    {
        // Base case
        if (node == null)
            return node;
  
        // Convert to DLL using bintree2listUtil()
        node = bintree2listUtil(node);
  
        // bintree2listUtil() returns root node of the converted
        // DLL.  We need pointer to the leftmost node which is
        // head of the constructed DLL, so move to the leftmost node
        while (node.left != null)
            node = node.left;
  
        return node;
    }
  
    /* Function to print nodes in a given doubly linked list */
    void printList(Node node)
    {
        while (node != null)
        {
            System.out.print(node.data + " ");
            node = node.right;
        }
    }
  
    /* Driver program to test above functions*/
    public static void main(String[] args)
    {
        BinaryTree tree = new BinaryTree();
  
        // Let us create the tree shown in above diagram
        tree.root = new Node(10);
        tree.root.left = new Node(12);
        tree.root.right = new Node(15);
        tree.root.left.left = new Node(25);
        tree.root.left.right = new Node(30);
        tree.root.right.left = new Node(36);
  
        // Convert to DLL
        Node head = tree.bintree2list(tree.root);
  
        // Print the converted list
        tree.printList(head);
    }
}


Python3
# Python program to convert
# binary tree to doubly linked list
 
class Node(object):
     
    """Binary tree Node class has
    data, left and right child"""
    def __init__(self, item):
        self.data = item
        self.left = None
        self.right = None
 
def BTToDLLUtil(root):
     
    """This is a utility function to
    convert the binary tree to doubly
    linked list. Most of the core task
    is done by this function."""
    if root is None:
        return root
 
    # Convert left subtree
    # and link to root
    if root.left:
         
        # Convert the left subtree
        left = BTToDLLUtil(root.left)
 
        # Find inorder predecessor, After
        # this loop, left will point to the
        # inorder predecessor of root
        while left.right:
            left = left.right
 
        # Make root as next of predecessor
        left.right = root
         
        # Make predecessor as
        # previous of root
        root.left = left
 
    # Convert the right subtree
    # and link to root
    if root.right:
         
        # Convert the right subtree
        right = BTToDLLUtil(root.right)
 
        # Find inorder successor, After
        # this loop, right will point to
        # the inorder successor of root
        while right.left:
            right = right.left
 
        # Make root as previous
        # of successor
        right.left = root
         
        # Make successor as
        # next of root
        root.right = right
 
    return root
 
def BTToDLL(root):
    if root is None:
        return root
 
    # Convert to doubly linked
    # list using BLLToDLLUtil
    root = BTToDLLUtil(root)
     
    # We need pointer to left most
    # node which is head of the
    # constructed Doubly Linked list
    while root.left:
        root = root.left
 
    return root
 
def print_list(head):
     
    """Function to print the given
       doubly linked list"""
    if head is None:
        return
    while head:
        print(head.data, end = " ")
        head = head.right
 
# Driver Code
if __name__ == '__main__':
    root = Node(10)
    root.left = Node(12)
    root.right = Node(15)
    root.left.left = Node(25)
    root.left.right = Node(30)
    root.right.left = Node(36)
 
    head = BTToDLL(root)
    print_list(head)
 
# This code is contributed
# by viveksyngh


C#
using System;
 
// C# program to convert binary tree to double linked list
 
/* A binary tree node has data, and left and right pointers */
public class Node
{
    public int data;
    public Node left, right;
 
    public Node(int item)
    {
        data = item;
        left = right = null;
    }
}
 
public class BinaryTree
{
    public Node root;
    /* This is the core function to convert Tree to list. This function
       follows steps 1 and 2 of the above algorithm */
 
    public virtual Node bintree2listUtil(Node node)
    {
        // Base case
        if (node == null)
        {
            return node;
        }
 
        // Convert the left subtree and link to root
        if (node.left != null)
        {
            // Convert the left subtree
            Node left = bintree2listUtil(node.left);
 
            // Find inorder predecessor. After this loop, left
            // will point to the inorder predecessor
            for (; left.right != null; left = left.right)
            {
                ;
            }
 
            // Make root as next of the predecessor
            left.right = node;
 
            // Make predecessor as previous of root
            node.left = left;
        }
 
        // Convert the right subtree and link to root
        if (node.right != null)
        {
            // Convert the right subtree
            Node right = bintree2listUtil(node.right);
 
            // Find inorder successor. After this loop, right
            // will point to the inorder successor
            for (; right.left != null; right = right.left)
            {
                ;
            }
 
            // Make root as previous of successor
            right.left = node;
 
            // Make successor as next of root
            node.right = right;
        }
 
        return node;
    }
 
    // The main function that first calls bintree2listUtil(), then follows
    // step 3 of the above algorithm
 
    public virtual Node bintree2list(Node node)
    {
        // Base case
        if (node == null)
        {
            return node;
        }
 
        // Convert to DLL using bintree2listUtil()
        node = bintree2listUtil(node);
 
        // bintree2listUtil() returns root node of the converted
        // DLL.  We need pointer to the leftmost node which is
        // head of the constructed DLL, so move to the leftmost node
        while (node.left != null)
        {
            node = node.left;
        }
 
        return node;
    }
 
    /* Function to print nodes in a given doubly linked list */
    public virtual void printList(Node node)
    {
        while (node != null)
        {
            Console.Write(node.data + " ");
            node = node.right;
        }
    }
 
    /* Driver program to test above functions*/
    public static void Main(string[] args)
    {
        BinaryTree tree = new BinaryTree();
 
        // Let us create the tree shown in above diagram
        tree.root = new Node(10);
        tree.root.left = new Node(12);
        tree.root.right = new Node(15);
        tree.root.left.left = new Node(25);
        tree.root.left.right = new Node(30);
        tree.root.right.left = new Node(36);
 
        // Convert to DLL
        Node head = tree.bintree2list(tree.root);
 
        // Print the converted list
        tree.printList(head);
    }
}
 
// This code is contributed by Shrikant13


Javascript


C++
// A C++ program for in-place
// conversion of Binary Tree to DLL
#include 
using namespace std;
 
/* A binary tree node has data,
and left and right pointers */
class node {
public:
    int data;
    node* left;
    node* right;
};
 
/* This is the core function to convert
Tree to list.*/
void bintree2listUtil(node* root, node** head, node** tail)
{
    if (root == NULL)
        return;
 
    bintree2listUtil(root->left, head, tail);
 
    if (*head == NULL) {
        *head = root;
    }
 
    root->left = *tail;
 
    if (*tail != NULL) {
        (*tail)->right = root;
    }
 
    *tail = root;
 
    bintree2listUtil(root->right, head, tail);
}
 
// The main function that first calls
// bintree2listUtil()
node* bintree2list(node* root)
{
    // Base case
    if (root == NULL)
        return root;
 
    node* head = NULL;
    node* tail = NULL;
 
    bintree2listUtil(root, &head, &tail);
 
    return head;
}
 
/* Helper function that allocates a new node with the
given data and NULL left and right pointers. */
node* newNode(int data)
{
    node* new_node = new node();
    new_node->data = data;
    new_node->left = new_node->right = NULL;
    return (new_node);
}
 
/* Function to print nodes in a given doubly linked list */
void printList(node* node)
{
    while (node != NULL) {
        cout << node->data << " ";
        node = node->right;
    }
}
 
/* Driver code*/
int main()
{
    // Let us create the tree shown in above diagram
    node* root = newNode(10);
    root->left = newNode(12);
    root->right = newNode(15);
    root->left->left = newNode(25);
    root->left->right = newNode(30);
    root->right->left = newNode(36);
 
    // Convert to DLL
    node* head = bintree2list(root);
 
    // Print the converted list
    printList(head);
 
    return 0;
}
 
// This code is contributed by rathbhupendra


Java
// A Java program for in-place
// conversion of Binary Tree to DLL
import java.util.*;
 
class GFG{
 
/* A binary tree node has data,
and left and right pointers */
static class node {
 
    int data;
    node left;
    node right;
};
 
    /*
     * This is the core function to convert Tree to list.
     */
static node head, tail;
static void bintree2listUtil(node root)
{
    if (root == null)
        return ;
 
    node left = root.left;
    node right = root.right;
 
    bintree2listUtil(root.left);
 
    if (head == null) {
        head = root;
    }
 
    root.left = tail;
 
    if (tail != null) {
        (tail).right = root;
    }
 
    tail = root;
 
    bintree2listUtil(root.right);
}
 
// The main function that first calls
// bintree2listUtil()
static void bintree2list(node root)
{
   
    // Base case
    if (root == null)
        head= root;
 
   
 
    bintree2listUtil(root);
 
    }
 
/* Helper function that allocates a new node with the
given data and null left and right pointers. */
static node newNode(int data)
{
    node new_node = new node();
    new_node.data = data;
    new_node.left = new_node.right = null;
    return (new_node);
}
 
    /* Function to print nodes in a given doubly linked list */
static void printList()
{
    while (head != null) {
        System.out.print(head.data+ " ");
        head = head.right;
    }
}
 
    /* Driver code */
public static void main(String[] args)
{
   
    // Let us create the tree shown in above diagram
    node root = newNode(10);
    root.left = newNode(12);
    root.right = newNode(15);
    root.left.left = newNode(25);
    root.left.right = newNode(30);
    root.right.left = newNode(36);
    head = null;
    tail = null;
   
    // Convert to DLL
    bintree2list(root);
 
    // Print the converted list
    printList();
 
}
}
 
// This code is contributed by umadevi9616


C#
// A C# program for in-place
// conversion of Binary Tree to DLL
using System;
 
public class GFG {
 
    /*
     * A binary tree node has data, and left and right pointers
     */
public    class node {
 
    public    int data;
    public    node left;
    public    node right;
    };
 
    /*
     * This is the core function to convert Tree to list.
     */
    static node head, tail;
 
    static void bintree2listUtil(node root) {
        if (root == null)
            return;
 
        node left = root.left;
        node right = root.right;
 
        bintree2listUtil(root.left);
 
        if (head == null) {
            head = root;
        }
 
        root.left = tail;
 
        if (tail != null) {
            (tail).right = root;
        }
 
        tail = root;
 
        bintree2listUtil(root.right);
    }
 
    // The main function that first calls
    // bintree2listUtil()
    static void bintree2list(node root) {
 
        // Base case
        if (root == null)
            head = root;
 
        bintree2listUtil(root);
 
    }
 
    /*
     * Helper function that allocates a new node with the given data and null left
     * and right pointers.
     */
    static node newNode(int data) {
        node new_node = new node();
        new_node.data = data;
        new_node.left = new_node.right = null;
        return (new_node);
    }
 
    /* Function to print nodes in a given doubly linked list */
    static void printList() {
        while (head != null) {
            Console.Write(head.data + " ");
            head = head.right;
        }
    }
 
    /* Driver code */
    public static void Main(String[] args) {
 
        // Let us create the tree shown in above diagram
        node root = newNode(10);
        root.left = newNode(12);
        root.right = newNode(15);
        root.left.left = newNode(25);
        root.left.right = newNode(30);
        root.right.left = newNode(36);
        head = null;
        tail = null;
 
        // Convert to DLL
        bintree2list(root);
 
        // Print the converted list
        printList();
 
    }
}
 
// This code is contributed by gauravrajput1


Javascript


输出
25 12 30 10 36 15 

另一种方法:
算法:

  1. 以有序的方式遍历树。
  2. 在访问每个节点时,跟踪 DLL 的头和尾指针,使用尾指针将每个访问的节点插入到 DLL 的末尾。
  3. 返回列表的头部。

下面是上述方法的实现:

C++

// A C++ program for in-place
// conversion of Binary Tree to DLL
#include 
using namespace std;
 
/* A binary tree node has data,
and left and right pointers */
class node {
public:
    int data;
    node* left;
    node* right;
};
 
/* This is the core function to convert
Tree to list.*/
void bintree2listUtil(node* root, node** head, node** tail)
{
    if (root == NULL)
        return;
 
    bintree2listUtil(root->left, head, tail);
 
    if (*head == NULL) {
        *head = root;
    }
 
    root->left = *tail;
 
    if (*tail != NULL) {
        (*tail)->right = root;
    }
 
    *tail = root;
 
    bintree2listUtil(root->right, head, tail);
}
 
// The main function that first calls
// bintree2listUtil()
node* bintree2list(node* root)
{
    // Base case
    if (root == NULL)
        return root;
 
    node* head = NULL;
    node* tail = NULL;
 
    bintree2listUtil(root, &head, &tail);
 
    return head;
}
 
/* Helper function that allocates a new node with the
given data and NULL left and right pointers. */
node* newNode(int data)
{
    node* new_node = new node();
    new_node->data = data;
    new_node->left = new_node->right = NULL;
    return (new_node);
}
 
/* Function to print nodes in a given doubly linked list */
void printList(node* node)
{
    while (node != NULL) {
        cout << node->data << " ";
        node = node->right;
    }
}
 
/* Driver code*/
int main()
{
    // Let us create the tree shown in above diagram
    node* root = newNode(10);
    root->left = newNode(12);
    root->right = newNode(15);
    root->left->left = newNode(25);
    root->left->right = newNode(30);
    root->right->left = newNode(36);
 
    // Convert to DLL
    node* head = bintree2list(root);
 
    // Print the converted list
    printList(head);
 
    return 0;
}
 
// This code is contributed by rathbhupendra

Java

// A Java program for in-place
// conversion of Binary Tree to DLL
import java.util.*;
 
class GFG{
 
/* A binary tree node has data,
and left and right pointers */
static class node {
 
    int data;
    node left;
    node right;
};
 
    /*
     * This is the core function to convert Tree to list.
     */
static node head, tail;
static void bintree2listUtil(node root)
{
    if (root == null)
        return ;
 
    node left = root.left;
    node right = root.right;
 
    bintree2listUtil(root.left);
 
    if (head == null) {
        head = root;
    }
 
    root.left = tail;
 
    if (tail != null) {
        (tail).right = root;
    }
 
    tail = root;
 
    bintree2listUtil(root.right);
}
 
// The main function that first calls
// bintree2listUtil()
static void bintree2list(node root)
{
   
    // Base case
    if (root == null)
        head= root;
 
   
 
    bintree2listUtil(root);
 
    }
 
/* Helper function that allocates a new node with the
given data and null left and right pointers. */
static node newNode(int data)
{
    node new_node = new node();
    new_node.data = data;
    new_node.left = new_node.right = null;
    return (new_node);
}
 
    /* Function to print nodes in a given doubly linked list */
static void printList()
{
    while (head != null) {
        System.out.print(head.data+ " ");
        head = head.right;
    }
}
 
    /* Driver code */
public static void main(String[] args)
{
   
    // Let us create the tree shown in above diagram
    node root = newNode(10);
    root.left = newNode(12);
    root.right = newNode(15);
    root.left.left = newNode(25);
    root.left.right = newNode(30);
    root.right.left = newNode(36);
    head = null;
    tail = null;
   
    // Convert to DLL
    bintree2list(root);
 
    // Print the converted list
    printList();
 
}
}
 
// This code is contributed by umadevi9616

C#

// A C# program for in-place
// conversion of Binary Tree to DLL
using System;
 
public class GFG {
 
    /*
     * A binary tree node has data, and left and right pointers
     */
public    class node {
 
    public    int data;
    public    node left;
    public    node right;
    };
 
    /*
     * This is the core function to convert Tree to list.
     */
    static node head, tail;
 
    static void bintree2listUtil(node root) {
        if (root == null)
            return;
 
        node left = root.left;
        node right = root.right;
 
        bintree2listUtil(root.left);
 
        if (head == null) {
            head = root;
        }
 
        root.left = tail;
 
        if (tail != null) {
            (tail).right = root;
        }
 
        tail = root;
 
        bintree2listUtil(root.right);
    }
 
    // The main function that first calls
    // bintree2listUtil()
    static void bintree2list(node root) {
 
        // Base case
        if (root == null)
            head = root;
 
        bintree2listUtil(root);
 
    }
 
    /*
     * Helper function that allocates a new node with the given data and null left
     * and right pointers.
     */
    static node newNode(int data) {
        node new_node = new node();
        new_node.data = data;
        new_node.left = new_node.right = null;
        return (new_node);
    }
 
    /* Function to print nodes in a given doubly linked list */
    static void printList() {
        while (head != null) {
            Console.Write(head.data + " ");
            head = head.right;
        }
    }
 
    /* Driver code */
    public static void Main(String[] args) {
 
        // Let us create the tree shown in above diagram
        node root = newNode(10);
        root.left = newNode(12);
        root.right = newNode(15);
        root.left.left = newNode(25);
        root.left.right = newNode(30);
        root.right.left = newNode(36);
        head = null;
        tail = null;
 
        // Convert to DLL
        bintree2list(root);
 
        // Print the converted list
        printList();
 
    }
}
 
// This code is contributed by gauravrajput1

Javascript


输出
25 12 30 10 36 15