hdu 6121 Build a tree

最新推荐文章于 2021-11-10 14:21:48 发布

原创最新推荐文章于 2021-11-10 14:21:48 发布 · 666 阅读

0 ·

CC 4.0 BY-SA版权

ACM-树专栏收录该内容

20 篇文章

订阅专栏

本文介绍了一种解决特定树结构问题的方法，通过将树结构分为三种节点类型，并使用递归方式计算节点值，最终实现对树结构的有效遍历。特别针对k=1的情况进行了特殊处理。

对于树的每一层，我们可以分为三种节点，①满节点的k叉树②不满的k叉树③比第一种情况少一层的满节点的k叉树，然后从叶子节点开始不断转移到上一层。

对于k=1进行特判，打表找规律。。。。真TM刺激。。。爽的不行不行的，感觉看了题解是很容易，但是自己去想，一步步不出错，还是要有一定的脑洞的

#include<bits\stdc++.h>
using namespace std;
typedef long long ll;
int main(){
int t;
scanf("%d",&t);
while(t--)
{
    ll n,k;
    cin>>n>>k;
    if(k==1)
    {
         ll tt=n%4;
           if(tt==1)printf("1\n");
           else if(tt==2)printf("%lld\n",n+1);
           else if(tt==3)printf("0\n");
           else printf("%lld\n",n);
    continue;
    }
    ll s,remain,a,b,c;
    a=b=c=0;
    s=k;
    ll sum=1,flag=0;
    while(sum+s<=n)
    {
        sum+=s;
      if((double)s>1e18/(double)k)//可能会越界，
            {
                flag=1;
                break;
            }
        s*=k;
    }
    remain=n-sum;
    ll vala,valb,valc;
    vala=k+1;
    valb=(remain%k)==0?0:remain%k+1;
    valc=1;
    a=remain/k;
    b=remain%k==0?0:1;
    if(flag==0)s/=k;
    c=s-a-b;
    ll ans=remain%2;
    while(a+b+c>1)
    {
        if(a%2)ans^=vala;
        if(b%2)ans^=valb;
        if(c%2)ans^=valc;
        if(a%k||c%k)b=1;
        valb=a%k*vala+(k-b-a%k)*valc+1+valb;
        a/=k;
        vala=k*vala+1;
        c/=k;
        valc=k*valc+1;
    }
    cout<<(ans^n)<<endl;
}
    return 0;
}