codeforces 792D - Paths in a Complete Binary Tree

最新推荐文章于 2020-07-18 23:58:53 发布

原创最新推荐文章于 2020-07-18 23:58:53 发布 · 491 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#树状数组 #codeforces

ACM - 数据结构专栏收录该内容

9 篇文章

订阅专栏

本文介绍了一种使用lowbit技巧进行二叉树遍历的算法，并通过对比宏定义与函数实现的方式展示了不同方法的性能差异。该算法适用于快速定位二叉树中节点的父节点及左右子节点。

#include<cstdio>
#include<iostream>
#define lowbit(x) x&(-x)
typedef long long ll;
using namespace std;
ll n,q,num,root;string s;
int main()
{
	scanf("%I64d%I64d",&n,&q);
	root=(n+1)/2;
	for(ll q_i=1;q_i<=q;q_i++)
	{
		cin>>num>>s;
		for(int step=0;step<s.size();step++)
		{
			ll lowbit_num=lowbit(num);//假设num这个节点是左子节点
			if(s[step]=='U' && num!=root)
			{
				ll num_u=num+lowbit_num;//求出在假设情况下的num的父节点num_u
				ll lowbit_num_u=lowbit(num_u);
				if(num_u - lowbit_num_u/2 == num) num=num_u;//如果根据父节点求出来的左孩子就是num，那么num确实是左子节点
				else num=num-lowbit_num;//否则num就是右子节点
			}
			if(s[step]=='L') num-=lowbit_num/2;
			if(s[step]=='R') num+=lowbit_num/2;
		}
		printf("%I64d\n",num);
	}
}

思路来自http://blog.youkuaiyun.com/Courage_kn/article/details/69218592

用#define比定义一个lowbit函数快……不过好像很多时候不能像函数那样随便用，容易出问题……