《算法笔记》9.3小节——数据结构专题(2)-＞树的遍历

9.3.1树的静态写法

9.3.2树的先根遍历

9.3.3树的层序遍历

9.3.4从树的遍历看DFS和BFS

DFS=先根遍历
BFS=层序遍历
PAT A1053Path of Equal Weight
见书上P306

#define _CRT_SECURE_NO_WARNINGS 1
#include <cstdio>
#include<cstring>
#include<queue>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn = 110;
struct node {
	int weight;
	vector<int> child;
}Node[maxn];
bool cmp(int a, int b)
{
	return Node[a].weight > Node[b].weight;
}
int n, m, S;//节点数，非叶子节点数，给定的和
int path[maxn];
void DFS(int index,int numNode,int sum)
{
	if (sum > S) return;
	if (sum == S)
	{
		if (Node[index].child.size() != 0) return;//说明不是叶子节点
		for (int i = 0; i < numNode; i++)
		{
			printf("%d",Node[path[i]].weight);
			if (i < numNode - 1) printf(" ");
			else
				printf("\n");
		}
	}
	for (int i = 0; i < Node[index].child.size(); i++)
	{
		int child = Node[index].child[i];
		path[numNode] = child;
		DFS(child,numNode+1,sum+Node[child].weight);
	}

}
int main()
{	
	scanf("%d%d%d",&n,&m,&S);
	for (int i = 0; i < n; i++)
	{
		scanf("%d",&Node[i].weight);
	}
	int id, k, child;
	for (int i = 0; i < m; i++)
	{
		scanf("%d%d",&id,&k);
		for (int j = 0; j < k; j++)
		{
			scanf("%d",&child);
			Node[id].child.push_back(child);
		}
		sort(Node[id].child.begin(), Node[id].child.end(), cmp);
	}
	path[0] = 0;
	DFS(0,1,Node[0].weight);
	return 0;
}

问题 A: 树查找
问题描述：有一棵树，输出某一深度的所有节点，有则输出这些节点，无则输出EMPTY。该树是完全二叉树。

• 输入

输入有多组数据。
每组输入一个n(1<=n<=1000)，然后将树中的这n个节点依次输入，再输入一个d代表深度。

• 输出

输出该树中第d层得所有节点，节点间用空格隔开，最后一个节点后没有空格。

• 样例输入

输出该树中第d层得所有节点，节点间用空格隔开，最后一个节点后没有空格。

• 样例输出

EMPTY
2 3

完全二叉树的数学题，这题应该放在上一节哇，一开始提交说编译错误，是因为没有加math.h头文件导致的

#define _CRT_SECURE_NO_WARNINGS 1
#include <cstdio>
#include<queue>
#include<vector>
#include<algorithm>
#include<math.h>
using namespace std;
const int maxn = 1010;
int tree[maxn];
int main()
{	
	int n;
	int d;
	int temp;
	while (scanf("%d", &n) != EOF)
	{
		if (n == 0)
			break;
		for (int i = 0; i < n; i++)
		{
			scanf("%d", &tree[i]);
		}
		scanf("%d", &d);
		temp = (int)pow(2.0, 1.0 * (d-1) )-1;
		if (n <= temp)
			printf("EMPTY\n");
		else
		{
			for (int i = temp ; i < n&&i< (int)pow(2.0, 1.0 * d)-1; i++)
			{
				printf("%d ",tree[i]);
			}
			printf("\n");
		}
	}	
	return 0;
}

问题 B: 树的高度
问题描述：一棵树有n个节点，其中1号节点为根节点。

• 输入

第一行是整数n，表示节点数
后面若干行，每行两个整数a b，表示b是a的子节点。

• 输出

求这棵树的高度（根节点为第1层）

• 样例输入

5
1 2
1 3
3 4
3 5

• 样例输出

3

一开始使用！=EOF来输入对应关系的，然后发现这样并不能终止，注意n个节点有n-1条边，用这个判断条件即可

#define _CRT_SECURE_NO_WARNINGS 1
#include <cstdio>
#include<queue>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn = 1010;
struct Tree {
	int id;
	int d;
}tree[maxn];
int fa[maxn][2];//对应关系
int maxd;
int n;
void DFS(int num,int temp) {
	if (maxd < tree[num].d)
		maxd = tree[num].d;
	for (int i = 0; i < n-1; i++)
	{
		if (fa[i][0] == num)
		{
			tree[fa[i][1]].d = tree[num].d + 1;
			DFS(fa[i][1], temp);
		}
			
	}
}
int main()
{	
	
	int i;
	int temp;
	tree[1].d = 1;//根节点的层数为1	
	while (scanf("%d", &n) != EOF)
	{
		//初始化
		for (i = 1; i <=n; i++)
			tree[i].id = i;
		tree[1].d = 1;

		for (i = 0; i < n - 1; i++)//!!!注意有多少个边
		{
			scanf("%d %d", &fa[i][0], &fa[i][1]);
		}
		
		DFS(1,temp);
		printf("%d\n",maxd);
	}	
	return 0;
}