E. Preorder(dfs思维)

LINK

题目

在这里插入图片描述

Examples

input
4
BAAAAAAAABBABAB
output
16
input
2
BAA
output
1
input
2
ABA
output
2
input
2
AAB
output
2
input
2
AAA
output
1

大致翻译:

给出一个n层一个完全二叉树以及它的前序遍历的结点上对应的字母,结点只能为A或B,现在可以任意翻转节点上的左右子树,形成一个新的前序遍历字符串,问字符串有多少种不同的情况。

思路:

将每个节点的左右子树构成的字符串按左小右大的顺序放置,这样可以排除AB\BA对结果的影响,
若此时结点的左右子树不同,说明左右交换可以改变字符串的顺序,统计不同情况的数目(设为y),答案即2y(每交换一次可以得到两种不同的字符串);

代码:

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int N=2e5+10;
const int inf=0x3f3f3f3f;
const int mod=998244353;
int n;
string s;
int ans=1; 
string dfs(int x){
	if(x<(1<<(n-1))){//非叶子节点 
		string l=dfs(x<<1);
		string r=dfs(x<<1|1);
		if(l!=r){
			ans=(ans*2)%mod;
		}
		if(l>r)swap(l,r);
		return s[x]+l+r;
	}else{//叶子节点,直接返回  
		if(s[x]=='A')return "A";
		else return "B";
	} 
}
int main() {
	ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	cin>>n>>s;
	s=" "+s;
	dfs(1);
	cout<<ans<<endl;
	return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
这段代码存在以下几个问题: 1. `root = BiTreeNode`应该改为`root = BiTreeNode()`,即需要调用`BiTreeNode`类的构造函数来创建一个具体的节点对象。否则,变量`root`仅仅是一个指向`BiTreeNode`类的引用,而不是一个具体的节点对象。 2. `createBiTree(order, self=None)`方法的定义中,`self`参数是没有用的,应该删除。 3. `preOrder(root)`, `inOrder(root)`和`postOrder(root)`这三个函数的调用方式不正确,应该使用`bt.preOrder(bt.root)`等方式来调用。 下面是更新后的代码: ```python class BiTreeNode: def __init__(self, data=None): self.data = data self.l_child = None self.r_child = None class BiTree(object): def __init__(self, root=None): self.root = root @classmethod def createBiTree(cls, order): q = LinkQueue() root = BiTreeNode() bt = BiTree(root) q.offer(root) for i in range(len(order)): c = order[i] node = q.peek() if node.l_child is None: newNode = BiTreeNode(c) node.l_child = newNode q.offer(newNode) elif node.r_child is None: newNode = BiTreeNode(c) node.r_child = newNode q.offer(newNode) q.poll() return bt def preOrder(self, root): if root is not None: print(root.data, end='') self.preOrder(root.l_child) self.preOrder(root.r_child) def inOrder(self, root): if root is not None: self.inOrder(root.l_child) print(root.data, end='') self.inOrder(root.r_child) def postOrder(self, root): if root is not None: self.postOrder(root.l_child) self.postOrder(root.r_child) print(root.data, end='') bt = BiTree.createBiTree('') print("先遍历:") bt.preOrder(bt.root) print() print("中遍历:") bt.inOrder(bt.root) print() print("后遍历:") bt.postOrder(bt.root) print() ``` 注意,上述代码假定`LinkQueue`类已经被定义并导入到当前模块。此外,由于没有输入实际的二叉树节点值,因此代码运行结果并不有意义。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值