leetcode 100 相同的樹

最近在刷leetcode ，順便記錄下來，就當筆記啦。

題目描述：

思路：先序遍歷，先對當前節點值進行判斷，然後在遞歸判斷左右子樹。

代碼如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool isSameTree(TreeNode* p, TreeNode* q) {
        if(p == NULL && q == NULL) return true;
        if(p == NULL && q != NULL) return false;
        if(p != NULL && q == NULL) return false;
        if(p -> val != q -> val) return false;
        if(isSameTree(p -> left,q -> left) == false) return false;
        if(isSameTree(p -> right,q -> right) == false) return false;
        return true;
    }
};

看了別人的解答，其實還可以稍微優化一下：

class Solution {
public:
    bool isSameTree(TreeNode* p, TreeNode* q) 
    {
        if (p == nullptr && q == nullptr) // 直接返回都爲空情況
        return true;
        else if (p == nullptr || q == nullptr)  // 因爲上一句話去除了都爲空情況，此句等同於 (p==nullptr && q!=nullptr)&&(p!=nullptr && q==nullptr)
            return false;
        else if (p->val != q->val)  // 當前子樹根節點的值不同
            return false;
        else 
            return isSameTree(p->left, q->left) && isSameTree(p->right, q->right);  // 兩個樹的左右孩子是否分別相等
    }
};