对于层次遍历二叉树,我们应该都挺熟悉。但是对于不同类型的返回值的函数,有些处理手段也不同。
1.例如,返回值是void型的话,此时如果打印二叉树的各个节点的值的话,直接在函数内部实现即可,而且遇到空指针时,可以在开头直接判断,类似于:
void levelOrder(TreeNode *T) { if(T == nullptr) return; ... }
2.但是如果返回时是一个数组或者是一个vector的话,此时如果实参是个空指针的话,就不能直接判断并且返回了,但可以忽略该节点的相关操作并且continue,继续下个循环,类似于:
vector<int> levelOrder(TreeNode *T) { vector<int> rs; ......... while(...) { if(T == nullptr) continue; } ... }
这可以看成是处理空指针的一个小技巧,《剑指offer》面试题32就可以这么用。