例题6-6 小球下落 (Dropping Balls, UVa 679) python实现

最新推荐文章于 2023-09-17 20:08:43 发布

ITSanta

最新推荐文章于 2023-09-17 20:08:43 发布

阅读量729

点赞数 1

分类专栏：蓝桥杯文章标签： python 算法数据结构二叉树

本文链接：https://blog.csdn.net/ITSanta/article/details/117136472

版权

蓝桥杯专栏收录该内容

2 篇文章 0 订阅

订阅专栏

本文介绍了如何使用Python解决UVa679（DroppingBalls）问题，该问题涉及二叉树和动态开关。给定一棵最大深度为D的二叉树，小球从节点1开始落下，遇到开关时状态会改变，影响小球的左右移动。代码实现了根据开关状态计算小球最终落在哪个叶子节点的功能，并在AcWing平台上通过了测试。

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

刘汝佳《算法竞赛入门经典》例题6-6 小球下落 (Dropping Balls, UVa 679) python实现

AcWing 1258 与本题基本一致，题目描述不太一样，内容是一样的，代码在AcWing中AC了。

有一棵二叉树，最大深度为D，且所有叶子的深度都相同。所有结点从上到下从左到右编号为1, 2, 3,…, 2D-1。在结点1处放一个小球，它会往下落。每个内结点上都有一个开关，初始全部关闭，当每次有小球落到一个开关上时，状态都会改变。当小球到达一个内结点时，如果该结点上的开关关闭，则往左走，否则往右走，直到走到叶子结点。
6-2

一些小球从结点1处依次开始下落，最后一个小球将会落到哪里呢？输入叶子深度D和小球个数I，输出第I个小球最后所在的叶子编号。假设I不超过整棵树的叶子个数。D≤20。输入最多包含1000组数据。

AcWing上的要求是

输入格式：

一行包含两个用空格隔开的整数 D 和 I。

输出格式：

对应输出第 I 个小球下落停止时的叶子序号。

数据范围

2≤D≤20,
1≤I≤524288

输入样例：

4 2

输出样例：

代码:

def fa(x,y,switch):
    if(switch == 0):
        x = x+1
        y = y*2
        switch = 1
    else:
        x = x+1
        y = y*2+1
        switch = 0
    return x,y,switch

import sys
try:
    while True:
        line=sys.stdin.readline().strip()
        if line == '':
            break
        k,n=map(int,line.split())
except:
    pass

lst=[]
for j in range(1,k+1):
    temp = []
    for i in range(2**(j-1),2**j-1+1): #-1是最后一个，但range不包括，所以需要+1
        temp.append(0)
    lst.append(temp)

x = 0
y = 0
y_last = 0
for p in range(n):
    x , y = 0 , 0
    for j in range(k):
        switch = lst[x][y]
        x_temp,y_temp,switch = fa(x,y,switch)
        lst[x][y] = switch
        y_last = y # 取最后一个数的时候需要y值。
        x,y = x_temp,y_temp
print([i for i in range(2**(k-1),2**k-1+1)][y_last]) #因为直接存的0，1，没有再改。