pat-玩转二叉树

来源:互联网 发布:股份公司股东人数 知乎 编辑:程序博客网 时间:2024/05/16 03:09

L2-011. 玩转二叉树

时间限制
400 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
陈越

给定一棵二叉树的中序遍历和前序遍历,请你先将树做个镜面反转,再输出反转后的层序遍历的序列。所谓镜面反转,是指将所有非叶结点的左右孩子对换。这里假设键值都是互不相等的正整数。

输入格式:

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

输出格式:

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

输入样例:
71 2 3 4 5 6 74 1 3 2 6 5 7
输出样例:
4 6 1 7 5 3 2

#include <iostream>#include<bits/stdc++.h>using namespace std;typedef struct node{    int data;    struct node*l;    struct node*r;}tree;tree* creat1(int xian[],int zhong[] ,int len){    if(len<=0)    return NULL;    tree*head = new tree;    head->data = *xian;    int i = 0;    for(;i<len;i++)    {        if(zhong[i]==*xian)        {            break;        }    }    head->l = creat1(xian+1,zhong,i);    head->r = creat1(xian+i+1,zhong+i+1,len-i-1);    return head;}void cccout(tree*root){    tree*p = NULL;    queue<tree*>g;    if(root)    {        g.push(root);    }    while(!g.empty())    {        p = g.front();        g.pop();        cout<<p->data<<" ";        if(p->l)        {            g.push(p->l);        }        if(p->r)        {            g.push(p->r);        }    }}void xxian(tree*root){    if(root)    {        cout<<root->data<<" ";        xxian(root->l);        xxian(root->r);    }}void ccout(tree*root){    queue<tree*>q;    tree*p =NULL;    if(root)    {      q.push(root);    }    int top = 1;    while(!q.empty())    {        p = q.front();        q.pop();        if(top)top =0;        else printf(" ");        cout<<p->data;        if(p->l)        {           q.push(p->l);        }         if(p->r)        {           q.push(p->r);        }    }}void swap(tree*root){    if(!root) return ;    swap(root->l);    swap(root->r);    swap(root->l,root->r);}//或者在层序遍历的时候倒过来int main(){    int n;    cin>>n;    int xian[36],zhong[36];    for(int i=0;i<n;i++)    {        cin>>zhong[i];    }    for(int i=0;i<n;i++)    {        cin>>xian[i];    }    tree*root = new tree;    root = creat1(xian,zhong,n);    swap(root);    ccout(root);    return 0;}静态数组版
#include  <iostream>#include  <cstdio>#include  <cstring>#include  <queue>using namespace std;const int maxn=1000000;int n,num(0);int pre[maxn],in[maxn],tree[maxn];void build(int tag,int left,int right){tree[tag]=pre[++num];if(left==right)return;int index=-1;for(int i=left;i<=right;i++){if(in[i]==tree[tag]){index=i;break;}}if(index==-1)return;if(index-1>=left)build(tag<<1,left,index-1);if(index+1<=right)build(tag<<1|1,index+1,right);}void bfs(){queue<int> que;que.push(1);int tag=tree[1];while(!que.empty()){int temp=que.front();que.pop();if(tree[temp]!=tag)printf(" ");printf("%d",tree[temp]);if(tree[temp*2+1]!=0)que.push(temp*2+1);if(tree[temp*2]!=0)que.push(temp*2);}}int main(){scanf("%d",&n);for(int i=1;i<=n;i++)scanf("%d",&in[i]);for(int i=1;i<=n;i++)scanf("%d",&pre[i]);build(1,1,n);bfs();printf("\n");return 0;}

 


0 0
原创粉丝点击