【題解】Luogu P5342 [TJOI2019]甲苯先生的線段樹

原題傳送門

挺有趣的一道題

\(c=1\)，暴力求出點權和n便可

\(c=2\)，先像\(c=1\)同樣暴力求出點權和n，考慮有多少路徑點權和也爲n

考慮設x爲路徑的轉折點，\(L\)爲\(x\)向左兒子走的長度，\(R\)爲\(x\)向右兒子走的長度。易知當\(L，R\)肯定時，有惟一的\(x\)對應

以\(x\)爲轉折點，\(L,R\)爲向左/右兒子走的距離，這時點權和至少爲\(Min=(2^{L+1}+2^{R+1}-3)x+2^R-1\)

此時x的取值必定珂以求出。考慮一下如何產生剩下\(n-Min\)的貢獻，這個貢獻必定是原來向左兒子走改爲向右兒子走所帶來的

咱們珂以進行記憶化搜索求出答案，記錄\(f[i][j][k]\)表示爲向左/右兒子走的距離爲\(i,j\)還差貢獻爲\(k\)的方案數（注意：答案要減去原來已有的那個解）

#include <bits/stdc++.h>
#define ll long long 
#define getchar nc
using namespace std;
inline char nc(){
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline ll read()
{
    register ll x=0,f=1;register char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
    return x*f;
}
inline void write(register ll x)
{
    if(!x)putchar('0');if(x<0)x=-x,putchar('-');
    static int sta[20];register int tot=0;
    while(x)sta[tot++]=x%10,x/=10;
    while(tot)putchar(sta[--tot]+48);
}
inline int Max(register int x,register int y)
{
    return x>y?x:y;
}
inline int dep(register ll n)
{
    int res=0;
    while(n)
        ++res,n>>=1;
    return res;
}
map<ll,ll> mp[55][55];
inline ll dfs(register int x,register int y,register ll z)
{
    if(x<y)
        x^=y^=x^=y;
    if(z<0||z>(2ll<<x)+(2ll<<y)-x-y-4)
        return 0;
    if(!x&&!y)
        return !z;
    if(mp[x][y].count(z))
        return mp[x][y][z];
    return mp[x][y][z]=dfs(x-1,y,z)+dfs(x-1,y,z-(1ll<<x)+1);
}
int T,d,c;
ll a,b,n,ans;
int main()
{
    T=read();
    while(T--)
    {
        n=0;
        d=read(),a=read(),b=read(),c=read();
        while(a!=b)
        {
            if(a>b)
                n+=a,a>>=1;
            else
                n+=b,b>>=1;
        }
        n+=a;
        if(c==1)
        {
            write(n),puts("");
            continue;
        }
        ans=0;
        if(dep(n)<=d)
            ++ans;
        for(register int l=1;l<=d;++l)
        {
            ll k=(2ll<<l)-1;
            if(k<=n&&dep(n/k)+l<=d)
                ans+=dfs(l,0,n%k);
        }
        for(register int l=1;l<=d;++l)
            for(register int r=1;r<=d;++r)
            {
                ll k=(2ll<<l)+(2ll<<r)-3,b=(1ll<<r)-1;
                if(k+b<=n&&dep((n-b)/k)+Max(l,r)<=d)
                    ans+=dfs(l-1,r-1,(n-b)%k);
            }
        write(ans-1),puts("");
    }
    return 0;
}