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

#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());
}
}