【信奥】收集树叶

题面

描述

给出一棵二叉树，收集并移除所有叶子，重复执行，直到树为空

输入

第一行一个正整数n，表示结点的个数

n行，第i行第一个数l表示第i个结点左孩子的序号，第二个数r表示第i个结点右孩子的序号，-1表示空

输出

一共有若干行

请按照叶子的收集顺序，每收集一批叶子，将他们的序号打印在一行内

同一行内叶子的打印顺序为从左向右

思路

先求出每一个节点的高，用桶存入数组，再输出即可

最终代码

#include<bits/stdc++.h>
 using namespace std;
 struct node
 {
 	int l, r;
 };
 unordered_map<int,node> tree;
 int n;
 int ans=INT_MIN;
 string res[10001];
 int geth(int root)
 {
 	if (root == -1) return 0;
 	int ll=geth(tree[root].l);
 	int rr=geth(tree[root].r);
 	int tmp = max(ll,rr)+1;
 	res[tmp]+=(to_string(root)+" ");
 	return tmp;
 }
 int main()
 {
 	cin >> n;
 	for (int i = 1; i <= n; i++)
 	{
 		scanf("%d %d", &tree[i].l, &tree[i].r);
 	}
 	int h=geth(1);
 	for(int i=1;i<=h;i++)
 	{
 		cout<<res[i]<<endl;
 	}
 	return 0;
 }