题意:
对于一颗给定的二叉树,我们的任务是反转它的所有奇数层的节点的权值 val(根节点所在层为第 0 层),操作完后返回根节点 root。
思路:
思路一:
dfs 镜像遍历:左子树按照左中右的顺序遍历 右子树按照右中左的顺序遍历
当遍历到 奇数层 的时候,根据镜像遍历的特点,可以发现:当前遍历的两个点就是需要对换的两个点,直接对换即可。
思路二:
bfs 提取遍历过程中的 各层的节点,如果当前层是 奇数层,则反转节点权值即可。
思路二的实现较为巧妙,具体可以参考我之前写的一篇关于 bfs 求二叉树宽度 的博客:二叉树专题–洛谷 P3884 [JLOI2009]二叉树问题。
由于我们只关注层数的奇偶性,因此我们只需设置一个只有 0、1 两种取值的变量 dep,如果为 0 表示偶数层,为 1 表示奇数层,
当奇偶性改变的时候,对变量 dep 异或一个 1 即可。
时间复杂度:
O ( n ) O(n) O(n)
代码:
代码一:dfs(很短)
/*** Definition for a binary tree node.* struct TreeNode {* int val;* TreeNode *left;* TreeNode *right;* TreeNode() : val(0), left(nullptr), right(nullptr) {}* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}* };*/class Solution {
public:void dfs(tn* a, tn* b, int dep){if(!a) return ;if(dep) swap(a->val, b->val);dfs(a->le, b->ri, dep ^ 1), dfs(a->ri, b->le, dep ^ 1);}TreeNode* reverseOddLevels(TreeNode* root) {dfs(root->le, root->ri, 1);return root;}
};
代码二:bfs(较长)
/*** Definition for a binary tree node.* struct TreeNode {* int val;* TreeNode *left;* TreeNode *right;* TreeNode() : val(0), left(nullptr), right(nullptr) {}* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}* };*/class Solution {
public:tn* bfs(tn* u){qu<tn*> q; q.push(u), q.push(npt);vc<tn*> level;int dep = 0;while(!q.EP()) {auto tt = q.FT(); q.pop();if(!tt){if(level.EP()) break;if(dep){for(int i = 0, j = level.SZ() - 1; i < j; ++i, --j) swap(level[i]->val, level[j]->val);}dep ^= 1;q.push(npt);level.clear();continue;}level.pb(tt);if(tt->left) q.push(tt->le), q.push(tt->ri);}return u;}TreeNode* reverseOddLevels(TreeNode* root) {return bfs(root);}
};
宏定义代码:
typedef pair<int, int> pii;
typedef long long ll;
typedef double db;
#define x first
#define y second
#define tn TreeNode
#define npt nullptr
#define le left
#define ri right
#define qu queue
#define dq deque
#define vc vector
#define pq priority_queue
#define umap unordered_map
#define uset unordered_set
#define EP() empty()
#define SZ() size()
#define FT() front()
#define pb push_back
#define pf push_front
#define pp pop_back