P1827 [USACO3.4]美国血统 American Heritage 题解

题目传送门

一、已知前序+中序,求后序

(1)、二叉树的遍历,就是一个深度优先搜索的过程:
前序

const int N = 1e6 + 10;
//树的结构体+存储数组
struct Node {
    int id;   // 当前结点ID
    int left; // 左结点ID
    int right;// 右结点ID
} t[N];
int n;

void pre_order(int x){
 printf("%d\n",x);
 if(t[x].left) pre_order(t[x].left);
 if(t[x].right) pre_order(t[x].right);
}

中序

void in_order(int x){
 if(t[x].left) in_order(t[x].left);
 printf("%d\n",x);
 if(t[x].right) in_order(t[x].right);
}

后序

void post_order(int x){
 if(t[x].left) post_order(t[x].left);
 if(t[x].right) post_order(t[x].right);
 printf("%d\n",x);
}

说白了,套路基本一样,就是三步:输出当前结点,递归左子树,递归右子树。根据输出顺序不同,产生了三种遍历方式。

算法步骤:
(1)给出一个前序遍历的字符串,那么首字母一定是子树的根。
(2)另外再给出同样子树的中序遍历字符串,那么通过上面查找到的首字母在中序遍历字符串中找到它,它的左边就是左子树,右边就是右子树。
(3)利用中序字符串左子树的范围和右子树的范围,就可以确定在前序字符串中左子树的范围和右子树的范围,通过递归就可以继续遍历左右子树。

很多二叉树的题目,不需要真正的还原创建出二叉树,只是根据字符串,不断的递归子串,在递归过程中输出想要的结果,这样更轻量一些,代码更短。当然,想要还原真正的二叉树也是完全可以的。

代码原理:

#include <bits/stdc++.h>
using namespace std;


string pre_str; //前序
string in_str;  //中序

/**
 * 功能:对二叉树进行后序遍历.根据四个参数,(1)确定子树的根,(2)确定左右子树的大小范围,(3)按后序进行输出递归
 * @param l1 前序遍历的起点
 * @param r1 前序遍历的终点
 * @param l2 中序遍历的起点
 * @param r2 中序遍历的终点
 */
void dfs(int l1, int r1, int l2, int r2) {
    if (l1 > r1 || l2 > r2) return;//规定边界条件
    int i = in_str.find(pre_str[l1]);                  //利用根左右的特性来在中序队列中查找
    int cnt = i - l2;//左子树的节点个数
    //前序:l1是根,左子树是从l1+1开始,到l1+cnt结束
    //中序:l2开始,到i-1结束
    dfs(l1 + 1, l1 + cnt, l2, i - 1);       //递归左子树
    dfs(l1 + cnt + 1, r1, i + 1, r2);          //递归右子树
    cout << pre_str[l1];    //输出根结点
}

/**
ABEDFCHG
CBADEFGH

参考结果:AEFDBHGC
*/

int main() {
    //输入中序遍历,前序遍历字符串
    cin >> in_str >> pre_str;
    int right = in_str.size() - 1;  //right索引:因为是0开始,所以要-1

    //递归构建还原二叉树
    //本题,并没有真正的进行构建二叉树,只是根据二叉树的特点,不断查找过程中,输出后序遍历的字符串
    dfs(0, right, 0, right);//每个字符串有两个指针指向头和尾
    return 0;
}

二、已知后序+中序,求前序

P1030 [NOIP2001 普及组] 求先序排列
https://www.luogu.com.cn/problem/P1030
姊妹题: https://www.cnblogs.com/littlehb/p/15088998.html

三、已知前序+后序,求中序个数

P1229 遍历问题
https://www.luogu.com.cn/problem/P1229
https://www.cnblogs.com/littlehb/p/15099843.html

posted @ 2021-08-02 08:23  糖豆爸爸  阅读(352)  评论(0编辑  收藏  举报
Live2D