洛谷UVA679 小球下落 Dropping Balls
题目:
许多的小球一个一个的从一棵满二叉树上掉下来组成一个新满二叉树,每一时间,一个正在下降的球第一个访问的是非叶子节点。然后继续下降时,或者走右子树,或者走左子树,直到访问到叶子节点。
决定球运动方向的是每个节点的布尔值。最初,所有的节点都是 FALSE,当访问到一个节点时,如果这个节点是 FALSE,则这个球把它变成 TRUE,然后从左子树走,继续它的旅程。如果节点是TRUE,则球也会改变它为 FALSE,而接下来从右子树走。满二叉树的标记方法如下图。
因为所有的节点最初为 FALSE,所以第一个球将会访问节点 1,节点 2 和节点 4,转变节点的布尔值后在在节点 8 停止。第二个球将会访问节点 1、3、6,在节点 12 停止。明显地,第三个球在它停止之前,会访问节点 1、2、5,在节点 10 停止。
现在你的任务是,给定新满二叉树的深度 d 和下落的小球的编号 i ,可以假定I不超过给定的新满二叉树的叶子数,写一个程序求小球停止时的叶子序号p。
输入样例 输出样例
4 2 12 3 4 7 10 1 512 2 2 3 8 128 255
思路:本题的意思是,所有的节点一开始都是false状态,关闭中。题目中的编号都是顺序编号,所以若当前节点为x,他的左孩子是2x,右孩子是2x+1.
当小球落下,则开启。当小球编号I为奇数,则落向左子树,它是往左子树走的第(I+1)/2个小球。
反之,若为偶数,则落向右子树,它是往右子树走的第I/2个小球。
代码中d是二叉树深度,I是编号,i是循环次数,ans则是最终答案。
下面是代码:
代码:
#include<bits/stdc++.h>
using namespace std;
int main()
{
int d, I;
cin >> d >> I;
int ans = 1;
for (int i = 1; i < d; i=i+1) {
if (I % 2 == 1)
ans = ans * 2;
else
ans = ans * 2 + 1;
if (I > 1)
I = (I + 1) / 2;
}
cout << ans;
}