开荒新系列之LeetCode: 004-Same Tree

开荒新系列之LeetCode: NO.004 Same Tree

在广图，连刷两道题好了。

题目大意

比较两个二叉树的异同，只有结构完全相同，且对应结点的值也相同的二叉树，才是完全一致的。

基本思路

啊，我的DFS和BFS，你们又要出现啦。

到底用哪种呢？

感觉更偏好于DFS一些。因为大部分二叉树在构造的时候都会偏向于平衡结构，因而BFS在抵达较深层次的时候，会使其任务队列过于庞大（最多需要2^h的大小，h为深度），对于平衡二叉树，DFS仅需要约2*h的大小的栈容量，当然如果是单边的二叉树，那为什么不用链表呢。

我的解法

写离线测试好麻烦，得想一想怎么构思一个离线测试集了

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
def dfs(self, root):
stack = [root]
while stack != []:
node_now = stack.pop()
yield node_now
# 采用右子树探寻
if node_now.left != None:
stack.append(node_now.left)
if node_now.right!= None:
stack.append(node_now.right)

def xor(self, a, b):
if a and b != None:
return (a or b) and (not a and not b)
if a == None and b == None:
return False
return True

def check(self,a, b):
if (a.val == b.val) and (self.xor(a.left, b.left) == False) and (self.xor(a.right, b.right) == False):
return True
return False

def isSameTree(self, p, q):
# 对空树处理
if (p and q)== None:
if (p or q) != None:
return False
return True

gtr_p = self.dfs(p)
gtr_q = self.dfs(q)
result = True
# 比较长度一致部分的结果
try:
while 1:
node_now_p = gtr_p.next()
node_now_q = gtr_q.next()
if not self.check(node_now_p, node_now_q):
return False#,node_now_p.val, node_now_q.val, self.check(node_now_p, node_now_q)
except: # stop Iteration....
return True #,node_now_p.val, node_now_q.val, self.check(node_now_p, node_now_q)

中间有不少语法错误，忘记怎么用了。

另外这个代码的问题是速度比较慢。

我猜测整个判断结构应该可以优化一下。