迭代+队列的方式,代码如下:
/**
* 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:
TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2)
{
if( nullptr == t1 )
return t2;
if( nullptr == t2 )
return t1;
t1->val += t2->val;
std::deque<TreeNode *> que1, que2;
TreeNode *Temp1 = nullptr, *Temp2 = nullptr;
que1.push_back(t1);
que2.push_back(t2);
while( false == que1.empty() && false == que2.empty() )
{
//此区域不能有入队操作
Temp1 = que1.front();
Temp2 = que2.front();
if( nullptr != Temp1->left && nullptr != Temp2->left ) //都存在时才在以后的迭代中检查其左子树的情况
{
Temp1->left->val += Temp2->left->val;
que1.push_back(Temp1->left);
que2.push_back(Temp2->left);
}
else
{
if( nullptr != Temp2->left ) //更改指针域,此处没有任何的入队操作,已经进行了合并操作,无需检查其左子树
{
Temp1->left = Temp2->left;
}
}
if( nullptr != Temp1->right && nullptr != Temp2->right )
{
Temp1->right->val += Temp2->right->val;
que1.push_back(Temp1->right);
que2.push_back(Temp2->right);
}
else
{
if( nullptr != Temp2->right )
{
Temp1->right = Temp2->right;
}
}
que1.pop_front();
que2.pop_front();
}
return t1;
}
};
既然这种题适合递归来解,那么还可以通过栈加迭代的方式来模拟递归的行为,我最开始想的是完全模拟,即将nullptr也入栈,并判断每次迭代的当前指向节点,具体思路是这样的,将nullptr节点入栈,发现是nullptr作出适当的反应,有一个需要注意的点是,此函数中的参数以及存到容器中的内容将会是右值,所以无法直接更改当前指向,还需要准备一个容器来存储当前被指向节点的父节点,当发现某次迭代的当前节点满足条件并为nullptr时,更改父节点指向,但还是有一个坑的,无法判断当前节点是父节点的哪一个子树,一个无法判断的典型例子:当前节点为nullptr,但实际情况是父节点的左右子树都为nullptr,此种情况下完全无法判断当前节点到底是父节点的左子树还是右子树,于是这种思路行不通,于是决定不将nullptr入栈。
决定使用第二种方式后,写到后来发现使用队列与栈在不将nullptr入栈的话其实都是一个意思,比较典型的一个例子是,t1是一个线性的每个子树都只有左子树的二叉树,t2是一个线性的每个子树都只有右子树的二叉树,于是便停止了栈方式的编写