PAT (Advanced Level) 1151——前序中序构造树+LCA+map ( Ps.数组开大点啊 )

题目传送门

终于遇到了LCA题，先复习一下LCA模板
题目中提到的key值取值范围在int之内，不能直接视为结点编号
我将preorder的index作为结点编号，即认为结点的前序遍历为1…n
同时使用map记录key值和结点编号的对应关系

一开始在建树的时候，没有想明白怎么给结点编号
竟然傻到用一个全局变量累加给结点编号，这样回溯的时候全局变量已经累加，就不对了

测试点4、5WA掉，竟是因为数组开的不够大（10000->10005）

其实这道题没有必要建树，我们只要得到结点的deep就可以了
学习一下柳神不用建树的姿势

建树

struct node {int lc;int rc;int x;
};
node T[N + 1];//按照preorder的index规定结点编号
int build(int pre1,int pre2,int in1,int in2,int dp) {int id = pre1;T[id].x = preorder[pre1];Tmap[preorder[pre1]] = id;      //key -> nodeT[id].lc = T[id].rc = -1;deep[id] = dp;if (pre1 == pre2) return id;int index = find(preorder[pre1], in1, in2);int left_sz = index - in1;int right_sz = in2 - index;if (left_sz > 0) {T[id].lc = build(pre1 + 1, pre1 + left_sz, in1, index - 1, dp + 1);f[T[id].lc][0] = id;}if (right_sz > 0) {T[id].rc = build(pre1 + left_sz + 1, pre2, index + 1, in2, dp + 1);f[T[id].rc][0] = id;}return id;
}

AC代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<map>
#include<cmath>
using namespace std;const int N = 10005;
map<int, int>Tmap;
int n, m, root;
int preorder[N + 1];
int inorder[N + 1];
int deep[N + 1];
int f[N][20], mj;int find(int x,int s,int t) {for (int i = s; i <= t; ++i)if (inorder[i] == x) return i;
}//按照preorder的index规定结点编号
void build(int pre1,int pre2,int in1,int in2,int dp) {int id = pre1;Tmap[preorder[pre1]] = id;      //key -> nodedeep[id] = dp;if (pre1 == pre2) return;int index = find(preorder[pre1], in1, in2);int left_sz = index - in1;int right_sz = in2 - index;int lc = pre1 + 1;int rc = pre1 + left_sz + 1;if (left_sz > 0) {build(pre1 + 1, pre1 + left_sz, in1, index - 1, dp + 1);f[lc][0] = id;}if (right_sz > 0) {build(pre1 + left_sz + 1, pre2, index + 1, in2, dp + 1);f[rc][0] = id;}
}int lca(int x, int y) {if (deep[x] < deep[y]) swap(x, y);int d = deep[x] - deep[y];if (d)for (int i = 0; i <= mj && d; i++, d >>= 1)if (d & 1)x = f[x][i];if (x == y) return x;for (int i = mj; i >= 0; i--)if (f[x][i] != f[y][i]) {x = f[x][i];y = f[y][i];}return f[x][0];
}int main()
{scanf("%d%d",&m,&n);for (int i = 1; i <= n; ++i) scanf("%d", &inorder[i]);for (int i = 1; i <= n; ++i) scanf("%d", &preorder[i]);build(1,n,1,n,1);mj = (int)(log(n) / log(2)) + 1;for (int i = 1; i <= mj; ++i)for (int j = 1; j <= n; ++j)f[j][i] = f[f[j][i - 1]][i - 1];for (int i = 1; i <= m; ++i) {int x, y;scanf("%d%d", &x, &y);map<int, int>::iterator node1 = Tmap.find(x);map<int, int>::iterator node2 = Tmap.find(y);if (node1 == Tmap.end() && node2 == Tmap.end()) {printf("ERROR: %d and %d are not found.\n",x,y);}else if (node1 == Tmap.end() && node2 != Tmap.end()) {printf("ERROR: %d is not found.\n", x);}else if (node1 != Tmap.end() && node2 == Tmap.end()) {printf("ERROR: %d is not found.\n", y);}else {int ans = lca(Tmap[x], Tmap[y]);if (ans == Tmap[x]) {printf("%d is an ancestor of %d.\n", x, y);}else if (ans == Tmap[y]) {printf("%d is an ancestor of %d.\n", y, x);}else {printf("LCA of %d and %d is %d.\n", x, y, preorder[ans]);}}}return 0;
}

PAT (Advanced Level) 1151——前序中序构造树+LCA+map ( Ps.数组开大点啊 )相关推荐

java前中后序_前序中序求后序的java算法
二叉树的前序.中序.后序遍历的定义: 前序遍历:对任一子树,先访问跟,然后遍历其左子树,最后遍历其右子树: 中序遍历:对任一子树,先遍历其左子树,然后访问根,最后遍历其右子树: 后序遍历:对任一子树, ...
pat根据中序遍历和先序遍历_[leetcode/lintcode 题解] 前序遍历和中序遍历树构造二叉树...
[题目描述] 根据前序遍历和中序遍历树构造二叉树. 在线评测地址: 九章算法 - 帮助更多中国人找到好工作,硅谷顶尖IT企业工程师实时在线授课为你传授面试技巧www.jiuzhang.com [样例 ...
二叉树题目 ----7 前序中序遍历构造二叉树
前序中序遍历构造二叉树思路在前序中找根结点根据根结点 + 中序,分成左右两棵子树根据子树长度,把前序分成左右两颗子树递归处理子树 /*** Definition for a binary t ...
LintCode-73.前序遍历和中序遍历树构造二叉树
前序遍历和中序遍历树构造二叉树根据前序遍历和中序遍历树构造二叉树. 注意事项你可以假设树中不存在相同数值的节点样例给出中序遍历:[1,2,3]和前序遍历:[2,1,3]. 返回如下的树: 2 ...
根据前序和中序构造整个二叉树
根据前序和中序构造整个二叉树前序遍历:根左右中序遍历:左根右先根据前序遍历,得到根节点3,然后看中序遍历,可以知道9是3的左孩子[15,20,7]是3的右孩子再去先序遍历的到9,再到中 ...
二叉树的构造（前序+中序）---（后序 + 中序）
二叉树的构造(前序+中序)-(后序 + 中序) 思路:要对前序+中序(后序+中序)的构建树的动态过程要了解,思路比较简单,在了解了这个过程之后,理解下面代码就容易了. 过程参考图: 前序 + 中序: ...
前序中序、中序后序以及前序后序构造二叉树
文章目录前序中序中序后序前序后序定义的树节点如下, class TreeNode {int val;TreeNode left;TreeNode right;TreeNode(int x) { ...
通过前序中序，中序后序，前序后序来构造一二叉树
前序中序,以及中序后序可以确定唯一一颗二叉树,而前序后续是可以有多种情况,因为仅仅前序后续是无法确定根节点左右子树的问题: 首先解决通过前序和中序来构建一颗二叉树的问题: 前序=根节点+左子树+右子树 ...
根据前序和后序遍历构造二叉树（附前序中序代码）
889. 根据前序和后序遍历构造二叉树前序和中序道理相同改变一些细节就可以了,在最后附上了前序中序的代码题目描述返回与给定的前序和后序遍历匹配的任何二叉树. pre 和 post 遍历中的值 ...

PAT (Advanced Level) 1151——前序中序构造树+LCA+map ( Ps.数组开大点啊 )

建树

AC代码

PAT (Advanced Level) 1151——前序中序构造树+LCA+map ( Ps.数组开大点啊 )相关推荐

最新文章

热门文章