Dropping Balls 小球下落

最新推荐文章于 2024-10-09 22:28:58 发布

慎独666

最新推荐文章于 2024-10-09 22:28:58 发布

阅读量121

点赞数

分类专栏： acm 文章标签：算法

本文链接：https://blog.csdn.net/m0_63305704/article/details/125321715

版权

acm 专栏收录该内容

98 篇文章 0 订阅

订阅专栏

这篇博客讨论了一个关于树形结构的问题，即DroppingBalls-UVA679。题目要求根据给定的树的深度和小球编号，确定小球最后落在哪个节点。初始的数组模拟方法因时间复杂度过高导致超时。作者提出了优化方案，通过奇偶性判断小球在树中的移动路径，避免了存储所有节点。AC代码实现了这一优化，有效地解决了大数据量下的计算问题。

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

Dropping Balls - UVA 679 - Virtual Judge (vjudge.net)

题意就是给你一个树，给你树的深度和第几个小球，问小球最终掉落会在哪个编号上面，掉落规则：每次往相反的地方掉

这个题看树的结点有个很重要的性质：

一个根节点是k两个子结点分别是k*2和k*2+1，还有一共有2^d-1个结点

这是一开始用数组模拟的代码，就是把每个节点存到数组里面，然后掉落

#include<iostream>
#include<cstring>
using namespace std;
const int N=1e7+10;
bool st[N];
int main(){
	int Q;
	cin>>Q;
	while(Q--)
	{
		memset(st,0,sizeof(st));
		int d,k,id;
		cin>>d>>k;
		int n=(1<<d)-1;
		for(int i=0;i<k;i++)
		{
			id=1;
			while(1)
			{
				st[id]=!st[id];
				if(st[id]) id=id*2;
				else id=id*2+1;
				if(id>n) break;			
			}
		}
		cout<<id/2<<"\n";
	}
	return 0;
}

很可惜，TLE了，因为刚刚说了把所有的结点相当于存到数组里面，而且结点个数位2^d-1，d的数据范围1000，那肯定是tle的呀，所以要换个思路

对于根节点来说，奇数号的小球落在左子树，偶数号的小球落在右子树，然后分批落，第一批落在第二层（根节点不算），第二批落在第三层...依次类推

然后落的顺序和类比于根节点往下落，一个小球落下，直接由第几个小球分别的往上走，根据刚刚的三个点之间的关系，然后可以根据奇偶性来看，根据上面的tle的代码一样的求编号就好

ac代码：

#include<iostream>
using namespace std;
int main(){
	int Q;
	cin>>Q;
	while(Q--)
	{
		int d,k;
		cin>>d>>k;
		int id=1;
		for(int i=0;i<d-1;i++)
		{
			if(k%2)
			{
				id*=2;
				k=(k+1)/2;
			}
			else
			{
				id=id*2+1;
				k/=2; 
			}
		}
		cout<<id<<"\n";
	}
	return 0;
}