励志用尽量少的代码做高效的表达。
根据输入构造二叉树,输出该二叉树的先序序列。二叉树共有N个节点,节点编号是1到N。约定1号节点是根节点。
输入格式:
第一行输入整数N。 接下来有N行,依次给出1到N节点的左孩子和右孩子。对于这N行中的每一行,有两个整数。第i(i=1, 2, …, N)行中,第一个整数指出左孩子的编号,第二个整数指出右孩子的编号。如果整数值为0,表示没有左孩子或右孩子。
输出格式:
输出一行,内容是二叉树的先序序列。节点编号之间用空格隔开,行末有1个空格。
输入样例:
6
2 5
3 4
0 0
0 0
0 6
0 0
输出样例:
1 2 3 4 5 6
思路分析:
本题的核心思路是:结构体数组模拟指针建树。
最开始尝试用指针构建二叉树解题,但做得不舒服,解不出来。仔细思考了一下,有两点原因。
(一)、指针建树的优点在于:在不知道节点数目的情况下, 我们仍然可以通过“动态扩充”建树, 但本题已经给了结点数,及每个左右子树,这种情况下如果再用指针建树,就相当于给我们一个条件我们没用。
(二)、题中说每行对应每个节点的操作,也就是说:并不是乱序给我们的,我们可以通过循环遍历的方式很轻松的给结构体数组赋值。
考虑到这两点,我将其改进为结构体数组建树,不过结构体中的左右“子树”为int型。同样可以采用递归构造。很快就AC了。
代码:(不算注释为18行)
#include<iostream>
using namespace std;
struct Node {
int l, r;
};
//先序遍历
void Fir(Node nodes[], int root) {
if(root == 0) return; //递归结束的标志
cout << root << " "; //先序遍历的三行核心代码,
Fir(nodes, nodes[root].l);
Fir(nodes, nodes[root].r);
}
int main() {
int n; cin >> n;
Node nodes[n+1];
for(int i = 1; i <= n; i++) cin >> nodes[i].l >> nodes[i].r; //输入某节点的左右子树值
Fir(nodes, 1); //约定1号节点是根节点
return 0;
}
收获:
1、结构体数组模拟指针建树。