• L2-006 树的遍历


    给定一棵二叉树的后序遍历和中序遍历,请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。

    输入格式:

    输入第一行给出一个正整数N(≤),是二叉树中结点的个数。第二行给出其后序遍历序列。第三行给出其中序遍历序列。数字间以空格分隔。

    输出格式:

    在一行中输出该树的层序遍历的序列。数字间以1个空格分隔,行首尾不得有多余空格。

    输入样例:

    7
    2 3 1 5 7 6 4
    1 2 3 4 5 6 7
    

    输出样例:

    4 1 6 3 5 7 2


    解题思路: 输入后序序列(左右根), 中序序列(左根右), 建立二叉树, 根据中序和后序遍历的特点, 递归建树 , 最后层序遍历 输出结果
    #include <iostream>
    #include <algorithm>
    #include <vector>
    #include <stack>
    #include <queue>
    #include <cstdio>
    #include <cstring>
    using namespace std;
    const int N = 1100;
    int a[N], b[N];	// a 中序遍历序列, b 后序遍历序列 
    int tree[N], sum[N];	// tree 暂时存放树中的结点, sum 层序遍历序列 
    
    // 构建二叉树 
    // n 为节点编号, la为中序序列的首元素
    // ra 为中序序列的尾元素, lb 为后序遍历序列的首元素
    // rb 为后序遍历序列的尾元素 
    void build(int n, int la, int ra, int lb, int rb)
    {
    	if(ra < la || rb < lb)
    		return ;
    	if(la == ra)
    	{
    		tree[n] = a[la];
    		return ;
    	}
    	for(int i = la; i <= ra; ++ i)
    	{
    		if(a[i] == b[rb])  
    		// 后序遍历的最右边的为根节点, 找到根节点在中序序列中的位置
    		// 该位置的左边为左子树, 右边为右子树 
    		{
    			tree[n] = b[rb];
                  // 以当前的i为界限, 一分为二建树 build(2 * n, la, i - 1, lb, lb + i - la - 1); // i左侧(左子树) build(2 * n + 1, i + 1, ra, lb + i - la, rb - 1);  // i右侧(右子树) break; } } } // 层序遍历 (dfs) void bfs(int s) { queue<int> Q; Q.push(s); int ans = 1; while(Q.size()) { int y = Q.front(); Q.pop(); if(tree[y] == 0) continue; sum[ans ++] = tree[y]; Q.push(2 * y); Q.push(2 * y + 1); } } int main() { int n; memset(tree, 0, sizeof(tree)); cin >> n; for(int i = 1; i <= n; ++ i) cin >> b[i]; for(int i = 1; i <= n; ++ i) cin >> a[i]; build(1,1,n,1,n); bfs(1); for(int i = 1; i <= n; ++ i) { if(i == n) cout << sum[i]; else cout << sum[i] << " "; } return 0; }

      

  • 相关阅读:
    Openstack 通过 SQLAlchemy-ORM 访问数据库
    ulimit -c unlimited
    ajax 调用后台接口示例
    读书有什么用——北漂18年(番外篇三)
    zTree点击文字勾选复选框
    深度剖析 | 基于大数据架构的BI应用
    深度剖析 | 基于大数据架构的BI应用
    AngularJS之对话框
    AngularJS之依赖注入(实例一)
    AngularJS之$watch方法(监控动作)
  • 原文地址:https://www.cnblogs.com/mjn1/p/10567131.html
Copyright © 2020-2023  润新知