2022天梯赛热身赛 浪漫侧影(什么原题)

题意: 给定一棵二叉树的中序和后序遍历,求左视图和右视图。意思是求每一层的第一个节点和最后一个节点,层次遍历就行了。套皮题。
思路:
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;
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值