第100题 相同的树——树的前序递归遍历

二叉树数据结构定义

struct TreeNode {
    int val;
    TreeNode* left;
    TreeNode* right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

我的代码:

bool isSameTree(TreeNode* p, TreeNode* q) {
    if (p == NULL && q == NULL) { // 当前两树的根节点均为空
        return true;
    }
    else if ((p == NULL && q != NULL) || (p != NULL && q == NULL)) { // 只有一个为空,则直接返回两者不相等
        return false;
    }
    return isSameTree(p->left, q->left) && isSameTree(p->right, q->right) && p->val == q->val; //要求左子树和右子树均相等且根节点值也相等
}

简洁代码:

bool isSameTree(TreeNode* p, TreeNode* q) {
    if (!p && !q ) { // 当前两树的根节点均为空
        return true;
    }
	if (!p || !q) { // 只有一个为空,则直接返回两者不相等,两者都为空的情况前面已经排除
        return false;
    }
    return isSameTree(p->left, q->left) && isSameTree(p->right, q->right) && p->val == q->val; //要求左子树和右子树均相等且根节点值也相等
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章