算法课第十七周作业 | Same Tree

来源:互联网 发布:爱淘宝领红包 编辑:程序博客网 时间:2024/06/02 01:28

写在前面:

这周老师给了期末考试模拟题,做模拟题去了忘记写作业(拖延症借口),刚好有一道模拟题跟leetcode上的一样,将解题思路总结如下。


题意解读:

给定两个树,判断这两个树是不是一模一样的树。


解题思路:

分析树的结构,树由TreeNode组成的,分别是一个顶点值val和2个TreeNode指针组成。

用分治的思想,先判断val是否一样,再判断两个左边部分和两个右边部分是否一样。

先判断TreeNode是否是NULL,如果不是,将左边部分和右边部分视为另外的树,采用迭代的方法,再次判断val和比较左右两部分。

如果两个左边的TreeNode是空,则判断右边的树是否一样即可,反之亦然。

当然,也有可能传进来的是空指针,这一步放在最前面。

记得我提交Sicily的时候没有加第一步空指针的判断,看来leetcode的测试数据集更完备一些。


代码:

class Solution {public:    bool isSameTree(TreeNode* p, TreeNode* q) {    if(p == NULL && q == NULL)    return true;    else if (p == NULL || q == NULL)    return false;        if(p->left == NULL && p->right == NULL && q->left == NULL && q->right == NULL)        {        if(p->val == q->val)        return true;        else        return false;        }        else if (p->val == q->val && p->left == NULL && p->right != NULL && q->left == NULL && q->right != NULL)        {        return isSameTree(p->right, q->right);        }        else if (p->val == q->val && p->left != NULL && p->right == NULL && q->left != NULL && q->right == NULL)        {        return isSameTree(p->left, q->left);        }        else if (p->val == q->val && p->left != NULL && p->right != NULL && q->left != NULL && q->right != NULL)        {        return (isSameTree(p->left, q->left) && isSameTree(p->right, q->right));        }        else        return false;    }};