C++ 将二叉树叶子结点从左往右顺序串连

关键代码：

//将叶子结点从左往右顺序串连
void linkLeafNode(BTNode *p,BTNode *&head,BTNode *&tail)
{
    if(p!=NULL)
    {
        if(p->lchild==NULL&&p->rchild==NULL)//判断是否叶子结点
        {
            if(head==NULL)
            {
                head=p;
                tail=p;
            }
            else
            {
                tail->rchild=p;
                tail=p;
            }
        }
        linkLeafNode(p->lchild,head,tail);
        linkLeafNode(p->rchild,head,tail);
    }
}

完整测试代码：

#include <bits/stdc++.h>
#define MaxSize 100
/*
 * Created by HarvestWu on 2018/06/22.
 */
using namespace std;

typedef char ElemType;

//定义二叉树结构
typedef struct BTNode
{
    ElemType data;
    struct BTNode *lchild;
    struct BTNode *rchild;
} BTNode;

//由先序遍历、中序遍历结果构建二叉树
BTNode *CreateBT(ElemType pre[],ElemType in[],int l1,int r1,int l2,int r2)
{
    BTNode *s;
    int i;
    if(l1>r1)return NULL;       //序列中无元素返回NULL
    s=(BTNode*)malloc(sizeof(BTNode));
    s->lchild=s->rchild=NULL;
    for(i=l2; i<=r2; ++i)
        if(in[i]==pre[l1])      //查找等于当前子树根的结点在in[]中的位置
            break;
    s->data=in[i];
    s->lchild=CreateBT(pre,in,l1+1,l1+i-l2,l2,i-1);
    s->rchild=CreateBT(pre,in,l1+i-l2+1,r1,i+1,r2);
    return s;       //当前子树处理完毕
}

//将叶子结点从左往右顺序串连
void linkLeafNode(BTNode *p,BTNode *&head,BTNode *&tail)
{
    if(p!=NULL)
    {
        if(p->lchild==NULL&&p->rchild==NULL)//判断是否叶子结点
        {
            if(head==NULL)
            {
                head=p;
                tail=p;
            }
            else
            {
                tail->rchild=p;
                tail=p;
            }
        }
        linkLeafNode(p->lchild,head,tail);
        linkLeafNode(p->rchild,head,tail);
    }
}

int main()
{
    BTNode *p,*head,*tail;
    ElemType pre[]= {'A','B','D','E','H','J','K','L','M','N','C','F','G','I'};
    ElemType in[]={'D','B','J','H','L','K','M','N','E','A','F','C','G','I'};
    
	//由先序遍历、中序遍历结果构建二叉树  
    p=CreateBT(pre,in,0,13,0,13);  

    cout<<"将叶子结点从左往右顺序串连:"<<endl;
    head=tail=NULL;
    linkLeafNode(p,head,tail);
    while(head!=NULL)
    {
        cout<<head->data;
        if(head=head->rchild)
            cout<<"->";
    }
    return 0;
}