hdu 5524 规律题

Subtrees

Accepts: 60

Submissions: 170

Time Limit: 2000/1000 MS (Java/Others)

Memory Limit: 131072/131072 K (Java/Others)

问题描述

一棵有N个节点的完全二叉树，问有多少种子树所包含的节点数量不同。

输入描述

输入有多组数据，不超过1000组.
每组数据输入一行包含一个整数N.(1≤N≤1018)(1\leq N\leq {10}^{18})(1≤N≤10​18​​)

输出描述

对于每组数据输出一行，表示不同节点数的子树有多少种.

输入样例

5
6
7
8

输出样例

3
4
3
5

分析：一颗完全二叉树，左右子树都会为完全二叉树，其中必然有一个最后一层是满的。对于满二叉树，仔细想一下，发现其不同结点子树就是它的层数。那么答案就是左边二叉树不同结点子树加上右边的再减去重复的。接下来就是考虑哪边是满二叉树，如果左边是满二叉树，这种情况比右边是满二叉树多一（因为多了一层）。然后计算另一边不是满二叉树的情况，如果最后一层剩下的节点数是奇数，那么直接把左右完全二叉树答案加起来就行，因为一边是奇，一边是偶，不会重叠的。但是如果剩下下的节点数是偶数，那就要考虑下重叠情况了，因为最后一层节点数是偶数的话，左边二叉树和右边二叉树第n-1层的节点会有相同的情况，这就需要除以2，同时答案减一直到 l 为奇数。具体情况看代码吧。

#include<iostream>
#include<cstdio>
#define LL long long
using namespace std;
LL f[66];
int main()
{
    f[0]=0;
    LL add=1;
    for(int i=1;i<65;i++){
        f[i]=f[i-1]+add;
        add<<=1;
    }
    LL n;
    while(cin>>n){
        LL nn=n;
        LL cnt=0;
        while(nn){
            nn>>=1;
            cnt++;
        }
        if(f[cnt]==n){
            cout<<cnt<<endl;
            continue;
        }

        LL ans=cnt-2;
        LL l=n-f[cnt-1],r=f[cnt]-n;
        if(l>=r){
            ans++;

        }
         while(!(l&1) && cnt--){
            l >>= 1;
        }

         // cout<<f[cnt]<<"..."<<ans<<"..."<<cnt<<endl;
        cout << ans + cnt - 1 << endl;
    }
    return 0;
}