vector向量数组,创建与遍历树

vector向量数组,创建与遍历树

题目背景

题目:列出叶结点

对于给定的二叉树,本题要求你按从上到下、从左到右的顺序输出其所有叶节点。

输入格式:
首先第一行给出一个正整数 N(≤10),为树中结点总数。树中的结点从 0 到 N−1 编号。随后 N 行,每行给出一个对应结点左右孩子的编号。如果某个孩子不存在,则在对应位置给出 “-”。编号间以 1 个空格分隔。

输出格式:
在一行中按规定顺序输出叶节点的编号。编号间以 1 个空格分隔,行首尾不得有多余空格。

输入样例:

8
1 -
- -
0 -
2 7
- -
- -
5 -
4 6

输出样例:

4 1 5

说明

vector是一个向量,里面可以存放不止一个的数。那么我们可以创建一个向量数组,数组下标代表父节点,该向量下的所有数代表字节点。如:vector<int>tree[100],tree[i]向量下的tree[i][j]代表父节点i其中一个子节点是tree[i][j]

代码及注释

#include<bits/stdc++.h>

using namespace std;

int far[100];//far[i]=j,代表i节点的父节点是j;如果j是-1,代表是根
int father;//father,通过far数组,找出根节点
int main(int argc, char *argv[]) {
	int n,i,t;
	vector<int>tree[100],ans;//tree[]是向量数组,存放树;ans是存放最后要输出的叶子节点。
	char ch;	
	cin>>n;
	for(i=0;i<30;i++)far[i]=-1;//先让所有节点无父节点,即每个节点是一棵树。
	for(i=0;i<n*2;i++){//输入n个节点的左右孩子,即有2n次输入
		cin>>ch;//用cin,直接避开回车和空格
		if(ch=='-')continue;//-表示当前孩子不存在,直接下一轮循环
		else {
			t=ch-'0';
			tree[i/2].push_back(t);//把当前孩子插入当前的父亲节点下
			far[t]=i/2;//保存当前孩子节点的父节点
			//father=i/2;
		}
	}
	while(far[father]!=-1)father=far[father];//找到根节点
	queue<int>q;//层次遍历用队列,
	q.push(father);
	while(q.size()){//层次遍历找到根节点
		t=q.front();
		q.pop();
		if(tree[t].size()==0)ans.push_back(t);
		else {
			for(int j=0;j<tree[t].size();j++){
				q.push(tree[t][j]);
			}
		}
	}
	for(i=0;i<ans.size();i++){//输出根节点
		cout<<ans[i];
		if(i==ans.size()-1)cout<<endl;
		else cout<<" ";
	}
	return 0;
}
  • 2
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值