题意: 给定一棵二叉树的中序和后序遍历,求左视图和右视图。意思是求每一层的第一个节点和最后一个节点,层次遍历就行了。套皮题。
思路:
1.我的笨法,由于每个数的值不一样,直接建的树,之后bfs即可,写的有点麻烦了,应该用深度的,写成2^i对应位置了,因为n才20.
2.大佬的妙法,直接根据中序和后序跑前序。因为根左右不就是层次遍历嘛,连树都不用建。
时间复杂度: O(能过)
代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
const int N = 1e6+10;
int n,m,k,T;
int a[N];
int va[N]; //中序
int vb[N]; //后序
map<int,int> l,r;
int get(int l1,int r1,int l2,int r2)
{
if(l1 > r1) return -1;
if(l1 == r1)
{
return va[l1];
}
int root = vb[r2];
int idx;
for(int i=l1;i<=r1;++i)
{
if(va[i] == root)
{
idx = i;
break;
}
}
int len = idx-1-l1+1;
l[root] = get(l1,idx-1,l2,l2+len-1);
r[root] = get(idx+1,r1,l2+len,r2-1);
return root;
}
void bfs()
{
queue<PII> q;
q.push({1,vb[n]});
while(q.size())
{
PII tmp = q.front(); q.pop();
int u = tmp.second;
int wh = tmp.first;
a[wh] = u;
if(l[u]>0)
{
q.push({wh*2,l[u]});
}
if(r[u]>0)
{
q.push({wh*2+1,r[u]});
}
}
}
void solve()
{
cin>>n;
for(int i=1;i<=n;++i) cin>>va[i];
for(int i=1;i<=n;++i) cin>>vb[i];
get(1,n,1,n);
// for(int i=1;i<=n;++i)
// {
// cout<<i<<":"<<l[i]<<" "<<r[i]<<endl;
// }
bfs();
vector<int> ans,res;
for(int i=0;i<n;++i)
{
for(int j=(1<<i);j<=(1<<(i+1))-1;++j)
{
if(a[j])
{
ans.push_back(a[j]);
break;
}
}
for(int j=(1<<(i+1))-1;j>=(1<<i);--j)
{
if(a[j])
{
res.push_back(a[j]);
break;
}
}
}
printf("R: "); for(int i=0;i<res.size();++i) {if(i) cout<<' '; cout<<res[i];}
printf("\n");
printf("L: "); for(int i=0;i<ans.size();++i) {if(i) cout<<' '; cout<<ans[i];}
}
signed main(void)
{
solve();
return 0;
}
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
const int N = 202;
int in[N];
int post[N];
vector<int> va[N]; //深度为i的点
int n,m,k,T;
void pre(int root,int l,int r,int dep)
{
// if(l == 0) return;
if(l > r) return ;
int idx;
for(int i=l;i<=r;++i) if(in[i] == post[root]) idx = i;
va[dep].push_back(in[idx]);
//不能用idx-1,如果相等也是巧合,那个不是真正的左子树树根位置
pre(root-1-(r-idx),l,idx-1,dep+1); //这里要用中序求右子树长度推后序中左子树树根的位置,否则会错.
pre(root-1,idx+1,r,dep+1);
}
void solve()
{
cin>>n;
for(int i=1;i<=n;++i) cin>>in[i];
for(int i=1;i<=n;++i) cin>>post[i];
pre(n,1,n,1);
printf("R:");
for(int i=1;;++i)
{
if(va[i].size())
{
cout<<" "<<va[i].back();
}
else break;
}
puts("");
printf("L:");
for(int i=1;;++i)
{
if(va[i].size())
{
cout<<" "<<va[i][0];
}
else break;
}
}
signed main(void)
{
solve();
return 0;
}