欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

天梯废物我是

程序员文章站 2022-04-15 18:58:16
给定一棵完美二叉树的后序遍历,求完美二叉树的层次遍历。只需要求出该树的左右儿子的数量即可#includeusing namespace std;typedef long long ll;int n;struct node{int lnum;int rnum;int sz;int val;}tr[N];int a[N];vectordep[N];void build(int u){tr[u].sz=1;...

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

#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;
}

本文地址:https://blog.csdn.net/qq_45961321/article/details/110290478

相关标签: 二叉树