题解Luogu P5342 [TJOI2019]甲苯先生的线段树

Posted 2021-12-09 yzhang-rp-inf

原题传送门

挺有趣的一道题

\(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;
}