HDU 6121 Build a tree(k叉树的子树大小相异)

http://acm.hdu.edu.cn/showproblem.php?pid=6121

题目大意:

给你一颗 n 个节点的完全 k 叉树,问你这棵树中所有子树结点个数的总异或值。

分析:

我们很容易看到于发现对于这样的一颗树 , 肯定是只有一颗子树不是满k叉树 , 知道这样后 , 这就很简单了;

我们可以找到不满的树的编号P , 那左边就要(P-1)棵满树 , 右边有(n-P)棵少一层的满树 , 我们在分析,如果左边的树的数量是偶数 , 那我们可以不用计算 , 奇数只要计算一颗;右边同样的道理:

还有一个问题就是k==1的情况:需要找到规律

#include<bits/stdc++.h>
using namespace std;
#define ll long long
ll n,k;
///返回1~x层的满树有多少结点
ll sum(ll x)
{
    if(x<=0) return 0;
    if(x==1) return 1;
    ll ans=1;
    for(ll i=1 ; i<x ; i++)
    {
        ans*=k;ans++;
    }
    return ans;
}
///对于一颗深度为 x 层的满树,返回它的异或值
ll comf(ll x)
{
    if(x<=0) return 0;
    if(x==1) return 1;
    ll ans=1;
    if(k%2==0)
    {
       return sum(x);
    }
    else
    {
        ll temp=1;
        for(ll i=1 ; i<=x ; i++)
        {
            temp*=k;
            temp++;
            ans=ans^temp;
        }
    }
    return ans;
}
///对于一颗 k 叉树,返回他的异或值
ll F(ll n)
{
    //printf("520");
    if(n==1) return 1;

    ll temp=n-1;///找到不满的树
    ll L,R;
    ll x=1;///层数
    while((temp-1)/k>0)///防止爆long long
    {
        temp=(temp-1)/k;
        x++;
    }
    x++;
    if(n==sum(x)) return comf(x);
    L=temp-1;///满k的树有多少棵
    R=k-temp;///缺一层满k的树有多少课
    ll ans=n;
    if(L%2==1)
    {
        ans=ans^comf(x-1);
    }
    if(R%2==1)
    {
        ans=ans^comf(x-2);
    }
    ///不满k树有多少课节点
    n=n-L*sum(x-1)-R*sum(x-2)-1;
    ans=ans^F(n);
    return ans;

}
int main()
{
    int T;scanf("%d",&T);
    while(T--)
    {
        scanf("%lld%lld",&n,&k);
        if(k==1)
        printf("%lld\n",n%4==1?1:(n%4==2?n+1:(n%4==3?0:n)));
        else printf("%lld\n",F(n));
    }
}
View Code

猜你喜欢

转载自www.cnblogs.com/shuaihui520/p/10808359.html