hdu 6121 Build a tree（k叉树的异或和）

Build a tree

题目链接：Build a tree

官方题解：

从最下层开始模拟到最上层就可以了。

除了最底层之外，每一层最多有三种节点，
第一种是以这个节点为根的树是满k叉树，第二种是不满的k叉树，第三种也是满k叉树，不过比第一种少了一层。

sz1表示第一种节点的子树大小（不过一旦没有了第一种节点，它就表示第二种节点），sum1表示有多少个这种节点
sz2，sum2，sz3，sum3同理

因此，边模拟边异或即可

代码：

#include<bits/stdc++.h>
using namespace std;

typedef long long LL;
LL n,k;

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%lld%lld",&n,&k);
        if(k==1)//对于链来说要特判，否则直接模拟会TLE
        {
            switch(n%4)
            {
            case 1:
                printf("1\n");
                break;
            case 2:
                printf("%lld\n",n+1);
                break;
            case 3:
                printf("0\n");
                break;
            case 0:
                printf("%lld\n",n);
                break;
            }
        }
        else
        {
            LL tot,ans=0,sz1=0,sz2=0,sz3=0,sum1=0,sum2=0,sum3=0;
            for(tot=1; n; tot*=k)//找到最底层
            {
                n-=tot;
                if(n<tot*k)
                    break;
            }
            if(n)
            {
                sum1=n,sz1=1;
                if(n&1)
                    ans=1;//最后一层的结果
            }
            for(; tot>0; tot/=k)//从倒数第二层开始
            {
                if(sum1<k)
                    sz1=sum1*sz1+sz2+(k-sum1-sum2)*sz3+1,sum1=1,sz2=0,sum2=0;
                else
                {
                    if(sum1%k)
                        sz2=sum1%k*sz1+sz2+(k-sum1%k-sum2)*sz3+1,sum2=1;
                    else if(sum2)
                        sz2=sz2+(k-1)*sz3+1;
                    sz1=sz1*k+1;
                    sum1=sum1/k;
                }
                sz3=sz3*k+1;
                sum3=tot-sum1-sum2;
                if(sum1&1)
                    ans^=sz1;
                if(sum2&1)
                    ans^=sz2;
                if(sum3&1)
                    ans^=sz3;
            }
            printf("%lld\n",ans);
        }
    }
    return 0;
}