题目描述
判断一棵二叉树是否对称。注意,如果一个二叉树同此二叉树的镜像是同样的,定义其为对称的。
我用的深度遍历方法,将二叉树根节点的左右子树分别存储在一个向量中,递归完成之后比较两个向量是否相同,决定这是不是对称二叉树,经过测试,实验结果是正确的,但是在提交验证的时候不能通过,运行超时,可见额外的存储和比较还是比较耗时的。下面我把超时的代码跟不超时的代码贴出来,给大家比较一下,希望下次能够注意。
这是二叉树的结构体和构造函数
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
这一部分是超时的代码
class Solution {
public:
vector<int> vecleft;
vector<int> vecright;
vector<int> vec;
vector<int> dfs(TreeNode* node) {
if (node->left) dfs(node->left);
vec.push_back(node->val);
if (node->right) dfs(node->right);
return vec;
}
bool isSymmetrical(TreeNode* pRoot)
{
int leftsize = 0, rightsize = 0;
int test = 0;
if (pRoot == nullptr)return true;
if ((!pRoot->left&&pRoot->right) || (pRoot->left && !pRoot->right))return false;
vecleft = dfs(pRoot->left);
//这个地方一定要清空遍历之后的中间存储向量,不清空的话下次赋值会留下上次的值
vec.clear();
vecright = dfs(pRoot->right);
leftsize = vecleft.size();
rightsize = vecright.size();
if (leftsize != rightsize)return false;
for (int i = 0; i < leftsize; i++) {
//这个地方必须逆序比较,两次递归得到的结果正好是相反的
if (vecleft[i] == vecright[leftsize - i - 1]) {
test++;
}
}
if (test < leftsize)return false;
return true;
}
};
下面这一部分是不超时,完全通过的代码
class Solution {
public:
bool compair(TreeNode *Lnode,TreeNode *Rnode){
if(!Lnode&&!Rnode)return true;
if((!Lnode&&Rnode)||(Lnode&&!Rnode)||(Lnode->val!=Rnode->val))return false;
return Lnode->val==Rnode->val&&compair(Lnode->left, Rnode->right)&&compair(Lnode->right, Rnode->left);
}
bool isSymmetrical(TreeNode* pRoot)
{
if(!pRoot)return true;
return compair(pRoot->left, pRoot->right);
}
};
下面代码为main函数,可以针对上面代码进行测试
测试的时候注意加入头文件和命名空间。
#include<iostream>
#include<vector>
#include<malloc.h>
using namespace std;
int main() {
Solution s1;
bool sigal;
TreeNode *node = new TreeNode(9);
node->left = new TreeNode(6);
node->right = new TreeNode(6);
node->left->left = new TreeNode(5);
node->left->right = new TreeNode(7);
node->right->left = new TreeNode(7);
node->right->right = new TreeNode(5);
sigal = s1.isSymmetrical(node);
cout << sigal << endl;
}