层次遍历二叉树时的一个技巧

  对于层次遍历二叉树,我们应该都挺熟悉。但是对于不同类型的返回值的函数,有些处理手段也不同。

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就可以这么用。

原文地址:https://www.cnblogs.com/wangkundentisy/p/8635511.html