STL学习笔记vector向量数组，创建与遍历树

Posted 2021-11-05 Keep--Silent

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