leetcode226

来源:互联网 发布:宣城太守知不知全诗 编辑:程序博客网 时间:2024/06/14 05:54

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]);}
0 0
原创粉丝点击