1、Invert Binary Tree
Invert a binary tree.
4
/ \
2 7
/ \ / \
1 3 6 9
to
4
/ \
7 2
/ \ / \
9 6 3 1
一说交换就想起来刚学c的时候两个变量交换的算法,交换二叉树也是一样,就是把每个节点的左右节点换一下,那就通过递归让每个节点都做根节点,通过swap函数,交换根节点的左右节点,函数写好了直接用就好啦,还是递归到节点为空时退出。先左先右都无所谓,swap函数放在第一句也是可以的。
/**
* 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* invertTree(TreeNode* root) {
if(root){
invertTree(root->left);
invertTree(root->right);
swap(root->left,root->right);
}
return root;
}
};
1、swap()在c++11标准中扩大了适用范围,不仅可以交换任何类型的变量,还可以交换任何变量在函数作用后的值,还可以用于数组内容的交换。
template <class T> void swap (T& a, T& b)
{
T c(std::move(a)); a=std::move(b); b=std::move(c);
}
template <class T, size_t N> void swap (T &a[N], T &b[N])
{
for (size_t i = 0; i<N; ++i) swap (a[i],b[i]);
}