1102. Invert a Binary Tree (25)

本文介绍了一个关于二叉树的数据结构实现,包括如何根据输入构建二叉树、找到根节点、翻转二叉树,并进行层序和中序遍历。展示了具体的C++代码实现。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

    唯一注意一点,就是根结点这里并未给出,需要自己根据输入判断出来。

#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
#include <cmath>
#include <queue>

using namespace std;

#define INF (~(1<<31))

struct Node{
	int id, left, right;

	Node(){}
	Node(int i, int l, int r) : id(i), left(l), right(r){}
};

vector<Node> nodes;
vector<int> level;
vector<int> inorder;

void invert(int root){
	if(root == -1) return;

	swap(nodes[root].left, nodes[root].right);
	invert(nodes[root].left);
	invert(nodes[root].right);
}

void inorderTranversal(int root){
	if(root == -1) return;

	inorderTranversal(nodes[root].left);
	inorder.push_back(root);
	inorderTranversal(nodes[root].right);
}

int main(){
	int n;
	cin >> n;

	
	vector<int> father(n, -1);
	nodes.resize(n);

	for(int i = 0; i < n; ++i){
		char l, r;
		cin >> l >> r;
		int left = -1, right = -1;
		
		if(l != '-') left = l - '0';
		if(r != '-') right = r - '0';

		nodes[i] = Node(i, left, right);

		if(left != -1) father[left] = i;
		if(right != -1) father[right] = i;
	}

	int root = find_if(begin(father), end(father), [](int val){ return val == -1; }) - begin(father);

	invert(root);

	queue<int> Q;
	Q.push(root);

	while(!Q.empty()){
		int r = Q.front(); Q.pop();

		level.push_back(r);
		if(nodes[r].left != -1) Q.push(nodes[r].left);
		if(nodes[r].right != -1) Q.push(nodes[r].right);
	}

	inorderTranversal(root);

	for(size_t i = 0; i < level.size(); ++i){
		if(i) printf(" ");
		printf("%d", level[i]);
	}

	printf("\n");
	for(size_t i = 0; i < inorder.size(); ++i){
		if(i) printf(" ");
		printf("%d", inorder[i]);
	}

	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值