一直很怕有关树的题,做了几个之后发现也还好,要克服恐惧!!!
1.一直后序中序,求层序:
#include <iostream>
#include<vector>
using namespace std;
vector<int>c(999999,-1);
int n,in[33],post[33];
void getpost(int root,int start,int end,int index)
{
if(start>end)return ;
int i=start;
while(i<=end&&in[i]!=post[root])i++;
c[index]=post[root];
getpost(root-end+i-1,start,i-1,index*2+1);
getpost(root-1,i+1,end,index*2+2);
}
int main()
{
cin>>n;
for(int i=0;i<n;i++)cin>>post[i];
for(int i=0;i<n;i++)cin>>in[i];
getpost(n-1,0,n-1,0);
int cnt=0;
for(int i=0;i<c.size();i++)
{
if(c[i]!=-1&&cnt!=n-1)
{
cout<<c[i]<<" ";
cnt++;
}
else if(c[i]!=-1)
{
cout<<c[i]<<endl;
}
}
return 0;
}
2.已知前序中序,求非叶子节点左右子树调换后的层遍历:
#include <iostream>
#include<vector>
using namespace std;
int n,in[33], pre[33];
vector<int>c(100000,-1);
//int index;
void getpost(int root,int start,int end,int index)
{
if(start>end)return ;
int i=start;
while(i<end&&in[i]!=pre[root])i++;
c[index]=pre[root];
getpost(root+1,start,i-1,2*index+2);
getpost(root+i-start+1,i+1,end,2*index+1);
}
int main()
{
cin>>n;
for(int i=0;i<n;i++)cin>>in[i];
for(int i=0;i<n;i++)cin>>pre[i];
getpost(0,0,n-1,0);
int cnt=0;
for(int i=0;i<100000;i++)
{
if(c[i]!=-1&&cnt!=n-1)
{
cout<<c[i]<<" ";
cnt++;
}
else if(c[i]!=-1)
{
cout<<c[i]<<endl;
}
}
return 0;
}