剑指Offer - 九度1385 - 重建二叉树
2013-11-23 23:53
-
题目描述:
-
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并输出它的后序遍历序列。
-
输入:
-
输入可能包含多个测试样例,对于每个测试案例,
输入的第一行为一个整数n(1<=n<=1000):代表二叉树的节点个数。
输入的第二行包括n个整数(其中每个元素a的范围为(1<=a<=1000)):代表二叉树的前序遍历序列。
输入的第三行包括n个整数(其中每个元素a的范围为(1<=a<=1000)):代表二叉树的中序遍历序列。
-
输出:
-
对应每个测试案例,输出一行:
如果题目中所给的前序和中序遍历序列能构成一棵二叉树,则输出n个整数,代表二叉树的后序遍历序列,每个元素后面都有空格。
如果题目中所给的前序和中序遍历序列不能构成一棵二叉树,则输出”No”。
-
样例输入:
-
8 1 2 4 7 3 5 6 8 4 7 2 1 5 3 8 6 8 1 2 4 7 3 5 6 8 4 1 2 7 5 3 8 6
-
样例输出:
-
7 4 2 5 8 6 3 1 No
题意分析:
给定一棵二叉树的前序遍历和中序遍历,计算出对应的后序遍历。如果给定的数据不是有效的遍历,则输出“No”。
前序遍历:根-左子树-右子树
中序遍历:左子树-根-右子树
后序遍历:左子树-右子树-根
因此,前序遍历的第一个节点为根节点,根据根节点的值在中序遍历中找到其对应位置,左边是左子树,右边是右子树,然后左右递归求解即可。
需要注意的是:输出后序遍历时,根节点最后输出,所以递归求解时,也应该在左右子树递归完成之后才处理根节点。
由于每次找出在中序遍历中找根节点的位置需要O(n)的查找时间,O(1)处理根节点,于是有T(n) = 2 * T(n / 2) + O(n) + O(1),推导下复杂度:
T(n) = 2 * T(n / 2) + O(1) + O(n) T(n) = 2 * T(n / 2) + O(n)
T(n) = 4 * T(n / 4) + 2 * (O(n / 2)) + O(n)
T(n) = 4 * T(n / 4) + 2 * O(n)
T(n) = 2 ^ log(n) * T(1) + log(n) * O(n)
T(n) = O(n) + O(n * log(n))
T(n) = O(n * log(n))
递归求解,因为每个节点都会被递归到,所以空间复杂度为O(n)。
1 // 651379 zhuli19901106 1385 Accepted 点击此处查看所有case的执行结果 1024KB 1256B 0MS 2 // 201311142119 3 #include <cstdio> 4 #include <vector> 5 using namespace std; 6 7 void construct_tree(vector<int> &a, vector<int> &b, vector<int> &c, int l1, int r1, int l2, int r2) 8 { 9 if(l1 > r1 || l2 > r2){ 10 return; 11 } 12 13 if(r1 - l1 != r2 - l2){ 14 return; 15 } 16 17 int mid; 18 int nl, nr; 19 20 for(mid = l2; mid <= r2; ++mid){ 21 if(b[mid] == a[l1]){ 22 nl = mid - l2; 23 nr = r2 - mid; 24 if(nl > 0){ 25 construct_tree(a, b, c, l1 + 1, l1 + nl, l2, mid - 1); 26 } 27 if(nr > 0){ 28 construct_tree(a, b, c, r1 - nr + 1, r1, mid + 1, r2); 29 } 30 c.push_back(a[l1]); 31 return; 32 } 33 } 34 35 } 36 37 int main() 38 { 39 vector<int> a, b, c; 40 int n; 41 int tmp; 42 int i; 43 44 while(scanf("%d", &n) == 1){ 45 for(i = 0; i < n; ++i){ 46 scanf("%d", &tmp); 47 a.push_back(tmp); 48 } 49 for(i = 0; i < n; ++i){ 50 scanf("%d", &tmp); 51 b.push_back(tmp); 52 } 53 c.clear(); 54 construct_tree(a, b, c, 0, n - 1, 0, n - 1); 55 a.clear(); 56 b.clear(); 57 if(c.size() < n){ 58 printf("No\n"); 59 }else{ 60 for(i = 0; i < n; ++i){ 61 printf("%d ", c[i]); 62 } 63 printf("\n"); 64 } 65 c.clear(); 66 } 67 68 return 0; 69 }