这是二叉搜索树吗？(python)

vvva's'd

已于 2024-03-26 22:54:11 修改

阅读量257

点赞数 6

文章标签：数据结构算法

于 2024-03-26 22:53:14 首次发布

本文链接：https://blog.csdn.net/crknbkls/article/details/137060660

版权

文章讲述了如何通过编程实现判断整数序列是否符合二叉搜索树或其镜像的前序遍历规则，涉及Node类定义和深度优先搜索算法。

摘要由CSDN通过智能技术生成

这是二叉搜索树吗？

分数 25

全屏浏览

切换布局

作者陈越

单位浙江大学

一棵二叉搜索树可被递归地定义为具有下列性质的二叉树：对于任一结点，

其左子树中所有结点的键值小于该结点的键值；
其右子树中所有结点的键值大于等于该结点的键值；
其左右子树都是二叉搜索树。

所谓二叉搜索树的“镜像”，即将所有结点的左右子树对换位置后所得到的树。

给定一个整数键值序列，现请你编写程序，判断这是否是对一棵二叉搜索树或其镜像进行前序遍历的结果。

输入格式：

输入的第一行给出正整数 N（≤1000）。随后一行给出 N 个整数键值，其间以空格分隔。

输出格式：

如果输入序列是对一棵二叉搜索树或其镜像进行前序遍历的结果，则首先在一行中输出 YES ，然后在下一行输出该树后序遍历的结果。数字间有 1 个空格，一行的首尾不得有多余空格。若答案是否，则输出 NO。

输入样例 1：

7
8 6 5 7 10 8 11

输出样例 1：

YES
5 7 6 8 11 10 8

输入样例 2：

7
8 10 11 8 6 7 5

输出样例 2：

YES
11 8 10 7 5 6 8

输入样例 3：

7
8 6 8 5 10 9 11

输出样例 3：

NO

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

import sys
sys.setrecursionlimit(100000000)
class Node:
    def __init__(self, val=-1, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right
n = int(input())
l = list(map(int, input().split()))
if len(l) == 1:
    print("YES")
    print(*l)
    exit()
flag = False
# 判断有可能是镜像还是非镜像
if l[1] >= l[0]:
    flag = True
def dfs(left, right):
    global flag
    if left >= right:
        return
    if left + 1 == right:
        return Node(l[left])
    root = Node(l[left])
    # 找到比根节点小的区间和大的区间
    l_s,l_r = 1000000000,-1
    r_s,r_r = 1000000000,-1
    for i in range(left+1, right):
        if l[i] < root.val:
            l_s,l_r = min(l_s,i),max(l_r,i)
        else:
            r_s,r_r = min(r_s,i),max(r_r,i)
    # 镜像和非镜像走不同的递归 镜像左右孩子反过来指
    if flag == False:
        # 5 1 2 6 3 7 8 比根节点小的区间 (1,4) 
        # 大的区间(3,6) 区间有交叉不符合二叉搜索树
        if l_s != -1 and r_s != -1 and  r_s <= l_r:
            print("NO")
            exit()
        root.left = dfs(l_s,l_r+1)
        root.right = dfs(r_s,r_r+1)
    else:
        #同理
        if l_s != -1 and r_s != -1 and l_s <= r_r:
            print("NO")
            exit()
        root.right = dfs(l_s, l_r + 1)
        root.left = dfs(r_s, r_r + 1)
    return root
ans = []
def display(r):
    if r is None:return
    display(r.left)
    display(r.right)
    ans.append(r.val)
rt = dfs(0, n)
print("YES")
display(rt)
print(*ans)