由先序和中序序列,重建二叉树

 输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。

#include <stdio.h>
#include <iostream>
#include <string>
#include <vector>

using namespace std;

 struct TreeNode {
     int val;
     TreeNode *left;
     TreeNode *right;
     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 };


class Solution {
public:
    TreeNode* R(vector<int> a,int abegin,int aend,vector<int> b,int bbegin,int bend)
    {
        // 判断两个序列是否为空
        if(abegin>=aend || bbegin>=bend)
            return NULL;

        TreeNode* root=new TreeNode(a[abegin]);  // 某个子树的根

        // 遍历中序序列,找到根节点所在的index,也就是pivot,同一个元素在两个序列中位序的差
        int pivot;
        for(pivot=bbegin;pivot<bend;pivot++)
            if(b[pivot]==a[abegin])
                break;
        printf("pivot: %d\n", pivot);

        // 递归结构
        root->left=R(a, abegin+1, abegin+pivot-bbegin+1, b, bbegin, pivot);
        root->right=R(a, abegin+pivot-bbegin+1, aend, b, pivot+1, bend);

        return root;
    }

    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        return R(pre,0,pre.size(),vin,0,vin.size());
    }
};


void print(int ch, int level)
{
//    cout << ch << "在第" << level << "层" << endl;
    printf("%d_%d\t", ch, level);
}

void PreOrder(TreeNode* T, int level) {
    if(!T)
        return;

    print(T->val, level);
    PreOrder(T->left, level + 1);
    PreOrder(T->right, level + 1);
}

void InOrder(TreeNode* T, int level)
{
    if(!T)
        return;

    InOrder(T->left, level + 1);
    print(T->val, level);
    InOrder(T->right, level + 1);
}

void PostOrder(TreeNode* T, int level)
{
    if(!T)
        return;

    PostOrder(T->left, level + 1);
    PostOrder(T->right, level + 1);
    print(T->val, level);
}

int main() {
    vector<int> pre;
    int tmp[] = {1,2,4,7,3,5,6,8};
    for(int i=0; i< sizeof(tmp)/ sizeof(int); i++)
        pre.push_back(tmp[i]);

    int tmp2[] = {4,7,2,1,5,3,8,6};
    vector<int> vin;
    for(int i=0; i< sizeof(tmp2)/ sizeof(int); i++)
        vin.push_back(tmp2[i]);

    Solution s;
    TreeNode* pRoot = s.reConstructBinaryTree(pre, vin);

    printf("PreOrder:\t\t");
    PreOrder(pRoot, 1);
    printf("\n");

    printf("InOrder:\t\t");
    InOrder(pRoot, 1);
    printf("\n");

    printf("PostOrder:\t\t");
    PostOrder(pRoot, 1);
    printf("\n");

    return 0;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值