hdu-5573 Binary Tree

题目地址

hdu5573

题目大意

有一颗k层满二叉树，其中点的权值从1开始，为1，2，3… 现在每个点的权值可以加也可以减，让你找到一个路径使得最后总和为给定的n。

解题思路

这个题如果不看题目给出的条件的话确实挺难的，题目中一个非常重要的条件是N < 2^k, 那我们知道对于满二叉树来说，最左边的节点的权值为1，2，4，…2^k。我们将满二叉树最左边的节点加起来后和为sum，由于N < 2^k，因此N < sum。
那所有节点加起来后多出来的数值为sum - N，只需要把多出来的数值减去就好，又因为一个节点只能加或着减去，那我们减去了一个权值为m的节点，实际上对于sum来说是减去了2m。因此只需要减去（sum-N）/ 2就可以了。那问题就转换为了怎么样去凑（sum - N）/ 2。由于最左边的节点为1，2，4，…2^k ，其二进制分别为 0001, 0010, 0100, 1000…
这样的二进制通过组合是可以表示出2^k以内的任何，所以可以凑出来(sum - N) / 2。但需要注意的是直接计算（sum-N）/ 2的话，如果sum-N是奇数的话会丢掉一个1，那我们只需要在最后一个节点的时候走右边的路径就行。

具体是如何做的请阅读代码。

AC代码

#include <iostream>
#include <cstring>

using namespace std;

long long sum[62], xx[62];
bool vis[62];

void init()
{
	sum[1] = 1;
	
	for (int i=2; i<62; i++)
	{
		sum[i] = sum[i-1] * 2;
	}
}

int main()
{
	init();  //初始化，将最左边的节点值先保存下来
	
	int t, a, b;
	
	cin >> t;
	
	for (int q=1; q<=t; q++)
	{
		int temp = 0;
		
		cin >> a >> b;
		memset(vis, 0, sizeof(vis));
		
		for (int i=1; i<=b; i++)   //由于会更改sum数组，因此我们需要一个辅助数组
		{
			xx[i] = sum[i];
			temp += sum[i];   //计算最左边节点总和
		}
		
		temp = temp - a;
		
		if (temp & 1)  //结果为奇数，最后一个节点走右边
		{
			xx[b]++;
			temp++;
		}
		
		temp /= 2;
		
		for (int i=b; i>=1; i--)
		{
			if (xx[i] <= temp)  //temp > xx[i],说明二进制的这一位上的值为1，因此减去xx[i]
			{
				temp -= xx[i];
				vis[i] = 1;
			}
		}
		
		printf("Case #%d:\n", q);
		
		for (int i=1; i<=b; i++)
		{
			if (vis[i])
				printf("%d -\n", xx[i]);
			else
				printf("%d +\n", xx[i]);
		}
	}
	
	return 0;
}