[CF750G]New Year and Binary Tree Paths

题目大意

一颗无穷个节点的完全二叉树，编号满足线段树分配。
求有多少条树上的简单路径编号和为s？

结论

从根节点到一个编号为x的节点，到根编号和为2x-bitcount(x)。
用归纳法证明。
我们尝试枚举这条简单路径两个节点到lca分别的距离为a和b，设lca为x。
首先一定有$x*(2^{a+1}+2^{b+1}-3)$的编号和。
然后我们尝试把x子树所有编号都去掉x的贡献（不是减x，而是第一层-x，第二层-2x，第三层-4x，以此类推）。
容易发现新的这棵树中还满足一个编号为x的节点，到根编号和为2x-bitcount(x)。
然后我们知道$x*(2^{a+1}+2^{b+1}-3)+k=s$
k表示其他的贡献。
容易讨论k的取值范围在$[2^b-1,2^a-a+2^{b+1}-b-3]$。
讨论过程可以默认$a>0，b>0$然后发现a=0或b=0代入此式仍然是对的。最小值就是两边都往左最大值就是两边都往右。
k的最大值严格比$2^{a+1}+2^{b+1}-3$小，这意味着如果存在一个x能找到一个合法k，那么x+1或-1都不可能再有合法k（即$s-x*(2^{a+1}+2^{b+1}-3)$不在k的取值范围内）。
然后我们唯一确定这个x后，问题得到转化。
找到两个数$p<2^{a-1}以及q<2^b$
使得$2*(p+q)-bitcount(p)-bitcount(q)=s-x*(2^{a+1}+2^{b+1}-3)$
其中q的第b-1位必须是1。
然后我们可以枚举bitcount的和，这样就可以确定p+q的值。
于是我们进行数位DP，注意到p和q的上界刚好是2的次幂-1，所以这个数位DP不需要别的东西（如常见的和上界比较的维），然后因为和确定了，因此只要让每一位严格等于和的那一位即可，而不需要专门开一维记录，只需要用一维记录是否进位即可。

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int maxd=58;
ll f[maxd+10][maxd*2+10][2],two[maxd+10];
int i,j,k,l,t,n,m,a,b;
ll x,y,now,s,ans;
void work(int a,int b,ll now,int cnt){
    int i,j,k,l,r,t,s,d,mx=max(a-2,b-1)+2;
    fo(i,0,maxd+2)
        fo(j,0,cnt)
            f[i][j][0]=f[i][j][1]=0;
    f[0][0][0]=1;
    fo(i,0,mx-1){
        if ((now&two[i])) d=1;else d=0;
        fo(j,0,cnt)
            fo(t,0,1)
                if (f[i][j][t]){
                    fo(l,0,1)
                        fo(r,0,1){
                            if (l&&i>a-2) continue;
                            if (r&&i>b-1) continue;
                            if (!r&&i==b-1) continue;
                            s=l+r+t;
                            k=s/2;s=s%2;
                            if (s!=d) continue;
                            f[i+1][j+l+r][k]+=f[i][j][t];
                        }
                }
    }
    ans+=f[mx][cnt][0];
}
int main(){
    two[0]=1;
    fo(i,1,maxd+1) two[i]=two[i-1]*2;
    scanf("%I64d",&s);
    fo(a,0,maxd)
        fo(b,0,maxd){
            y=two[a+1]+two[b+1]-3;
            x=s/y;
            while (x){
                now=s-x*y;
                if (now>two[a]-a+two[b+1]-b-3){
                    x=0;
                    break;
                }
                if (now>=two[b]-1&&now<=two[a]-a+two[b+1]-b-3) break;
                x--;
            }
            if (!x) continue;
            now=s-x*y;
            fo(i,0,maxd*2)
                if ((now+i)%2==0) work(a,b,(now+i)/2,i);
        }
    printf("%I64d\n",ans);
}