第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; //要求左子樹和右子樹均相等且根節點值也相等
}