首页 > 代码库 > 剑指OFFER之二叉搜索树与双向链表(九度OJ1503)

剑指OFFER之二叉搜索树与双向链表(九度OJ1503)

题目描述:

输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。

 

输入:

输入可能包含多个测试样例。
对于每个测试案例,输入的第一行为一个数n(0<n<1000),代表测试样例的个数。
接下来的n行,每行为一个二叉搜索树的先序遍历序列,其中左右子树若为空则用0代替。

 

输出:

对应每个测试案例,
输出将二叉搜索树转换成排序的双向链表后,从链表头至链表尾的遍历结果。

 

样例输入:
1
2 1 0 0 3 0 0

 

样例输出:
1 2 3

解题思路:

  这道题应该是最近写的最繁琐的一道题了。

  首先输入按规则来,需要进行前序遍历输入

void createTree(BTree **b){  
    int m;
    scanf("%d",&m);
    if(m == 0)
        *b = NULL;
    else{
        BTree *s = (BTree *)malloc(sizeof(BTree));
        s->data =http://www.mamicode.com/ m;
        s->lchild = NULL;
        s->rchild = NULL;
        *b = s;
        createTree(&((*b)->lchild));
        createTree(&((*b)->rchild));
    }
}

  另外,整体的思路,是用一个指针记录转换的双链表表尾。

  每次进行中序遍历的转换。

  最先转换的是最左下的节点,

void converNode(BTree *b,BTree **p){
    if(b == NULL)
        return ;
    BTree *pnow = b;
    if(b->lchild != NULL)
        converNode(b->lchild,p);
    
    pnow->lchild = *p;
    if(*p != NULL)
        (*p)->rchild = pnow;

    *p = pnow;

    if(b->rchild != NULL)
        converNode(b->rchild,p);
}

  每次遍历节点的左孩子右孩子。把左孩子指向转换链表的尾节点,并把末尾指针的右孩子指向自己。右孩子指向节点的右孩子。如果没有右孩子就返回。下面是代码思路的步骤:

1 找到最左边也就是最小的节点,PLast = NULL;

 

2 让节点的左孩子指向链尾,然后链尾指针右移。如果右孩子为空就返回。

 

最后我们从尾遍历回头指针返回。

全部代码:

#include <stdio.h>
#include <stdlib.h>
typedef struct btree{
    int data;
    struct btree *lchild,*rchild;
}BTree;
 
void createTree(BTree **b);
void inorderTree(BTree *b);
BTree * convert(BTree *b);
void converNode(BTree *b,BTree **p);
 
int main(){
    int n;
    scanf("%d",&n);
    while(n--){
        BTree *b = (BTree *)malloc(sizeof(BTree));   
        createTree(&b);
        BTree *head = convert(b);
        while(head!=NULL){
            printf("%d ",head->data);
            head = head->rchild;
        }
        printf("\n");
    }
    return 0;
}
BTree* convert(BTree *b){
    BTree *pLast = NULL;
    converNode(b,&pLast);
 
    BTree *phead = pLast;
    while(phead != NULL && phead->lchild != NULL)
        phead = phead->lchild;
    return phead;
}
void converNode(BTree *b,BTree **p){
    if(b == NULL)
        return ;
    BTree *pnow = b;
    if(b->lchild != NULL)
        converNode(b->lchild,p);
     
    pnow->lchild = *p;
    if(*p != NULL)
        (*p)->rchild = pnow;
 
    *p = pnow;
 
    if(b->rchild != NULL)
        converNode(b->rchild,p);
}
void createTree(BTree **b){  
    int m;
    scanf("%d",&m);
    if(m == 0)
        *b = NULL;
    else{
        BTree *s = (BTree *)malloc(sizeof(BTree));
        s->data =http://www.mamicode.com/ m;
        s->lchild = NULL;
        s->rchild = NULL;
        *b = s;
        createTree(&((*b)->lchild));
        createTree(&((*b)->rchild));
    }
}
/**************************************************************
    Problem: 1503
    User: xhalo
    Language: C
    Result: Accepted
    Time:80 ms
    Memory:1704 kb
****************************************************************/