当前位置: 首页 > 工具软件 > 天梯 > 使用案例 >

天梯废物我是

朱祺
2023-12-01

给定一棵完美二叉树的后序遍历,求完美二叉树的层次遍历。
只需要求出该树的左右儿子的数量即可

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
	int n;
struct node{
	int lnum;
	int rnum;
	int sz;
	int val;
}tr[N];
int a[N];
vector<int>dep[N];
void build(int u)
{
	tr[u].sz=1;
	if((u*2)<=n)
	build(u<<1);
	if((u*2+1)<=n)
	build(u<<1|1);
	tr[u].lnum=tr[u<<1].sz;
	tr[u].rnum=tr[u<<1|1].sz;
	tr[u].sz=tr[u].lnum+tr[u].rnum+1;
}
void dfs(int u,int d,int l,int r)
{
	cout<<l<<" "<<r<<" "<<tr[u].lnum<<endl;
	if(l==r)
	{
		dep[d].push_back(a[l]);
		return ;
	}
	dep[d].push_back(a[r]);
	if(l<=l+tr[u].lnum-1)
	dfs(u*2,d+1,l,l+tr[u].lnum-1);
	if(l+tr[u].lnum<=r-1)
	dfs(u*2+1,d+1,l+tr[u].lnum,r-1);
}
int main()
{
	ios::sync_with_stdio();
	cin.tie(0);
	cout.tie(0);

	cin>>n;
	for(int i=1;i<=n;i++)
	cin>>a[i];
	build(1);
	dfs(1,1,1,n);
//	return 0;
	for(int i=1;;i++)
	{
		
		if(!dep[i].size())	return 0;
		for(int j=0;j<dep[i].size();j++)
		{
			int x=dep[i][j];
			if(j==dep[i].size()-1 && !dep[i+1].size())
			cout<<x<<endl;
			else
			cout<<x<<" ";
		}
	}
	return 0;
}

 类似资料: