思路
三种排序的基础是深度优先搜索(DFS),实现方式是递归(也就是栈)
三步:
1.确定递归函数的传入参数和返回值
这里我习惯写的是int 作为函数返回值,因为可以return 0;比较好控制。
我们可以通过传入的指针来获得树的节点,通过传入一个数组或者字符串 (指针或者引用)来得到遍历的结果。
所以函数名可以写为 int traval (TreeNode * current,. vector<int > &num){
函数体;
}
2.确定递归的终止条件
简单来说就是,挖到什么程度返回。显然,当向下的遍历指针current==NULL时,证明已经挖到底了,此时current不是叶子节点,不具备任何实际意义,此时直接return 0,不做任何处理。
所以在函数开始工作之前,先检查是否已经挖到底了:
if(current ==NULL){return 0;}
当它是叶子节点的时候,以中序遍历为例;
对于一棵树来说,其形状为:
左节点——中间节点———右节点
3.单层递归逻辑的实现:
返回中间节点 :num.push_back(current->val);————①
返回左节点: traval(current->left);————②
返回右节点: traval(current->left);————③
对于其他遍历顺序来说,只需要把123的顺序调换就能得到其他的遍历顺序。
以下为leetcode 144真题的答案:
public:
int traval(TreeNode* current, vector<int>& num) {
if (current == NULL) {
return 0;
} // 深度搜索,停止条件
num.push_back(current->val); // 返回中间节点--1
traval(current->left, num); // 返回左节点--2
traval(current->right, num); // 返回右节点--3
// 1,2,3调换顺序就可以实现前中后三种遍历顺序的递归实现。
return 0;
}
vector<int> preorderTraversal(TreeNode* root) {
vector<int> vec;
traval(root, vec);
return vec;
}