[实战演练]蜻蜓FM2014年校招笔试题目 - 规则二叉树

题目：某规则二叉树的定义是：对于树中任意两个叶结点A、B，他们与根结点的距离分别是d1和d2，|d1-d2|<=1。请写出函数 bool isRuledTree(Node *root)的代码实现。如果该二叉树为规则树，则返回true，否则返回false。

思路：其实就是比较所有叶结点的深度，最大值和最小值之间相差不超过1，则为规则树，可以遍历所有的结点，如果碰到叶结点，记录该叶结点的深度到数组或任意什么结构中，然后找到这些记录中的最大值和最小值，求差即可。在我的代码中，用一个栈结构存储叶结点的深度。代码实现如下：

typedef struct Node{

    int val;

    Node *right;

    Node *left;

} Node;



stack<int> record;



void isRuledTreeCore(Node *n, int depth){

    if(n->left == NULL && n->right == NULL){

        record.push(depth);

        return;

    }

    if(n->left != NULL)

        isRuledTreeCore(n->left,depth+1);

    if(n->right != NULL)

        isRuledTreeCore(n->right,depth+1);

    return;

}

bool isRuledTree(Node *root){

    if(root==NULL) return true;

    

    isRuledTreeCore(root,0);



    int big=0, small = INT_MAX;

    while(!record.empty()){

        int temp = record.top();

        if(temp>big) big = temp;

        if(temp<small) small = temp;

        record.pop();

    }

    if(big-small>=-1 && big-small<=1)

        return true;

    return false;

}

注意INT_MAX这个常量要#include<limits>，stack使用时要#include<stack>