(天梯赛日常坑队友)
知道中序和后序,输出层次遍历
AC代码:
1 #include<bits/stdc++.h> 2 using namespace std; 3 # define ll long long 4 # define inf 0x3f3f3f3f 5 const int maxn = 4e5+1000; 6 int a[maxn],b[maxn]; 7 struct node 8 { 9 int s; 10 node *left,*right; 11 node() 12 { 13 left=right=NULL; 14 } 15 }; 16 int cnt,flag=0; 17 node *build(node *root,int l,int r) 18 { 19 if(flag) 20 return NULL; 21 root=new node; 22 root->s=b[--cnt]; 23 int i; 24 int k=0; 25 for( i=l; i<r; i++) 26 { 27 if(a[i]==root->s) 28 { 29 k=1; 30 break; 31 } 32 } 33 if(k==0) 34 { 35 flag=1; 36 return NULL; 37 } 38 if(i<r-1) 39 root->right=build(root->right,i+1,r); 40 if(i>l) 41 root->left=build(root->left,l,i); 42 return root; 43 } 44 void print(node *root) 45 { 46 queue<node*>q; 47 int k=0; 48 q.push(root); 49 node *tmp=new node; 50 while(!q.empty()) 51 { 52 tmp=q.front(); 53 q.pop(); 54 if(tmp) 55 { 56 if(k==0) 57 { 58 k=1; 59 printf("%d",tmp->s); 60 } 61 else 62 printf(" %d",tmp->s); 63 q.push(tmp->left); 64 q.push(tmp->right); 65 } 66 } 67 } 68 int main() 69 { 70 int n; 71 scanf("%d",&n); 72 for(int i=0; i<n; i++) 73 scanf("%d",&b[i]); 74 for(int i=0; i<n; i++) 75 scanf("%d",&a[i]); 76 cnt=n; 77 node *root=NULL; 78 root=build(root,0,n); 79 print(root); 80 printf("\n"); 81 }
知道前序和中序,输出
AC代码:
1 node *build( node *root,int left,int right) 2 { 3 int i; 4 int t = a[cnt++]; 5 for( i = left ; i < right ; i++ ) 6 { 7 if( b[i] == t ) 8 break; 9 } 10 root = new node(); 11 root->index = t; 12 if( i > left && i < right ) 13 root->left = build(root->left,left,i); 14 if( i >= left && i <right-1 ) 15 root->right = build(root->right,i+1,right); 16 return root; 17 }
知道前序和后序,不太好确定一棵二叉树。