HDU 5524 Subtrees

由于要求一颗完全二叉树的子树的种类数 因为一棵完全二叉树当结点数已知时树的形状是固定的 所以可以通过递归去做 将n放进set之后 处理出n所对应的完全二叉树的左右子树的数目 递归访问即可 同时采用记忆化搜索的方式 如果set中已经存在改数 就不在访问 这样复杂度接近log2(n)²

对与当前的树的左右子树结点数的计算 可以先计算出最后一层有多少个结点 （预处理k层的满二叉树的结点数 二分查找 减去即可） 然后把最后一层大于上一层的数目分给右子树 左子树用（总-1-右）即可

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cctype>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<algorithm>
#include<set>
#define scnaf scanf
#define cahr char
#define bug puts("=========================");
using namespace std;
typedef long long ll;
const int maxn=3000+5;
const int mod=1000000007;
set<ll>st;
ll n;
ll a[200];
ll b[200];
int cnt=0;
void init()
{
ll z=1;
ll N=3e18;
for(int i=0; z<=N; i++,cnt++)
{
a[i]=z-1;
b[i]=z/2;
z*=2;
}
}
void dfs(ll n)
{
if(n==1) return;
int id=lower_bound(a,a+cnt,n)-a;
ll nn=n-a[id-1];
ll right=a[id-2]+max((ll)0,nn-b[id-1]);
ll left=n-right-1;
if(left>0&&st.count(left)==0)
{
st.insert(left);
dfs(left);
}
if(right>0&&st.count(right)==0)
{
st.insert(right);
dfs(right);
}
}
int main()
{
init();
while(~scanf("%I64d",&n))
{

st.clear();
st.insert(n);
dfs(n);
st.insert(1);
printf("%d\n",st.size());
}
}