hud5524 Subtrees 二叉树中不同节点数的字数个个数

用set储存子数的节点数，set中元素的数目及为结果

#include<cstdio>
#include<set>
#include<iostream>
using namespace std;
set<long long> s;
long long a[65];//每层节点的个数
void dfs(long long n,int D)//n为当前节点数，D为当前层数
{
	if(D==0 || n==0 || s.count(n)) return;
	s.insert(n);
	long long l;//最后一层左子树节点个数
	long long r;//最后一层右子树节点个数
	long long num=n-a[D]+1;//最后一层节点总数
	if(num<=a[D]/2)//右边满
	{
		l=num;
		r=0;
	}
	else//左边满或两边都满
	{
		l=a[D]/2;
		r=num-a[D]/2;
	}
	dfs(a[D-1]-1+l,D-1);
	dfs(a[D-1]-1+r,D-1);
}
int main()
{
#ifdef CDZSC
	freopen("in.in","r",stdin);
#endif
	a[1]=1;
	for(int i=2;i<65;i++)
		a[i]=a[i-1]*2;
	long long n;
	int D;
	while(~scanf("%I64d",&n))
	{
		s.clear();
		for(D=2;a[D]-1<n;D++)
		{}
		D--;
		dfs(n,D);
		printf("%d\n",s.size());
	}
}