hdu6121 Build a tree

链接

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

题解

这套题怎么全是打表找规律啊
我从最下面一层开始算，维护三种结点的数量，和三种结点的子树大小
然后累加
我一直$T$的飞起，发现原来是没考虑$k=1$
$k=1$的时候就是$1$一直异或到$n$
打表找规律发现它是有规律的

代码

#include <bits/stdc++.h>
using namespace std;
#define eps 1e-8
typedef long long ll;
ll read(ll x=0)
{
    ll c, f(1);
    for(c=getchar();!isdigit(c);c=getchar())if(c=='-')f=-f;
    for(;isdigit(c);c=getchar())x=x*10+c-48;
    return f*x;
}
ll cnt[100];
int main()
{
    ll T(read()), n, k, D, tot, t, a, b, c, i, j, sa, sb, sc, ans;
    while(T--)
    {
        n=read(), k=read();
        if(k==1)
        {
            if(n%4==0)cout<<n<<endl;
            if(n%4==1)cout<<1<<endl;
            if(n%4==2)cout<<n+1<<endl;
            if(n%4==3)cout<<0<<endl;
            continue;
        }
        tot=1;
        t=1;
        for(D=1;;D++)
        {
            if(n==tot)break;
            if(log(t)+log(k)>log(n-tot)-eps)
            {
                D++;
                break;
            }
            t*=k;
            tot+=t;
        }
        t=1;
        a=n, b=0, c=0;
        for(i=1;i<D;i++,t*=k)a-=t;
        sa=1, sb=0, sc=0;
        ans=0;
        cnt[1]=1;
        for(i=2;i<=D;i++)cnt[i]=cnt[i-1]*k;
        for(i=D;i;i--)
        {
            if(a&1)ans^=sa;
            if(b&1)ans^=sb;
            c=(cnt[i]-a-b);
            if(c&1)ans^=sc;
            if(a%k+c%k>0)sb=a%k*sa+sb+c%k*sc+1;
            sa=sa*k+1;
            sc=sc*k+1;
            if(b==0)b=!!(a%k);
            a/=k;
        }
        cout<<ans<<endl;
    }
    return 0;
}