中序序列

时间限制:C/C++ 2秒,其他语言4秒
空间限制:C/C++ 262144K,其他语言524288K
64bit IO Format: %lld
题目描述
给定一棵有n个结点的二叉树的先序遍历与后序遍历序列,求其中序遍历序列。
若某节点只有一个子结点,则此处将其看作左儿子结点

输入

5,[3,2,1,4,5],[1,5,4,2,3]

输出

[1,2,5,4,3]

思路:首先明确一点,如果题目最后不给 “若某节点只有一个子结点,则此处将其看作左儿子结点” 这句话,这题无解,根左右和左右根推不出左右树的分界点的。
可知这棵二叉树要么没有结点,要么必有左儿子。
每次先序遍历第一个位置必是根,而在后序遍历中,夹在这个根和最右点的片段便是右儿子。
上个ac的核心代码好了,懒得改了

class Solution {
    
    
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param n int 二叉树节点数量
     * @param pre intvector 前序序列
     * @param suf intvector 后序序列
     * @return intvector
     */
    vector<int> res;//存中序遍历的数组
    void deal(int l1,int r1,vector<int>& pre,int l2, int r2,vector<int>& suf )//定义变量先序遍历和后序遍历的各两个端点
    {
    
    
        if(l1>r1) return ;
        else if(l1==r1)    {
    
    res.push_back(pre[l1]);return ;}
        int pos=-1;
        for(int i=l2;i<=r2;i++)
        {
    
    
        if(pre[l1+1]==suf[i])//找左子树的根
        {
    
    
            pos=i;
            break;
        }
        }
//按照左根右的顺序调用递归
    deal(l1+1, l1+1+pos-l2,pre,l2,pos,suf);//先序遍历左端点和后序遍历两个端点已知,不难推得先序遍历的右端点
    res.push_back(pre[l1]);
    deal(r1+1-r2+1+pos,r1,  pre, pos+1,  r2-1,suf );//先序遍历右端点和后序遍历两个端点已知,不难推得先序遍历的左端点
        return ;
    }
    
    vector<int> solve(int n, vector<int>& pre, vector<int>& suf) {
    
    
        deal(0,n-1,pre,0,n-1,suf);
        return res;
    }
};

猜你喜欢

转载自blog.csdn.net/m0_53688600/article/details/113361886