Sum of Log(数位dp)

Sum of Log(数位dp)

传送门

题意：

求满足${\sum }_{i=0}^X{\sum }_{j=[i=0]}^Y[i\&j=0]\lfloor lo{g}_2(i+j)+1\rfloor$的值

思路：

数位dp：

我们可以用数位dp枚举[0,x]和[0,y]的。

二进制处理。

我们可以将其搞成二进制，那么可以通过数位dp得到$[i\&j=0]$，

其中$\lfloor lo{g}_2(i+j)+1\rfloor$,则会是$x=i|j$，即x最高位1所在的位数。

优化：

我们可以通过首先计算出的叶子节点个数来计算该节点贡献，即优化掉一维来统计答案。

代码：

#include <iostream>
#include <cstring>
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pll;
const ll mod=1e9+7;
const int N=33;
int t;
int x,y;
int a[N],b[N];
pll dp[N][2][2][2];
pll dfs(int p,int pre,int lim1,int lim2){
    if(p==0){
        return pll(1,0);
    }
    if(dp[p][pre][lim1][lim2].second!=-1) return dp[p][pre][lim1][lim2];
    int up1=lim1? a[p]:1;
    int up2=lim2? b[p]:1;
    ll ans=0, cnt=0;
    pll ret;
    for(int i=0;i<=up1;i++){
        for(int j=0;j<=up2;j++){
            if(!(i&j)){
                if((pre&&(i|j)==1)){
                    pll it=dfs(p-1,pre&&((i|j)==0),lim1&&(i==up1),lim2&&(j==up2));
                    ans=(ans+it.first*p+it.second)%mod;
                    cnt=(cnt+it.first)%mod;
                    //ans=(ans+dfs(p-1,num+1,lim1&&(i==up1),lim2&&(j==up2)))%mod;
                }else{
                    pll it=dfs(p-1,pre&&((i|j)==0),lim1&&(i==up1),lim2&&(j==up2));
                    ans=(ans+it.second)%mod;
                    cnt=(cnt+it.first)%mod;
                    //ans=(ans+dfs(p-1,num+i+j,lim1&&(i==up1),lim2&&(j==up2)))%mod;
                }
            }
        }
    }
    return dp[p][pre][lim1][lim2]=pll(cnt,ans);
}
ll solve(int num1,int num2){
    memset(dp,-1,sizeof dp);
    memset(a,0,sizeof a);
    memset(b,0,sizeof b);
    int tot1=0,tot2=0;
    while(num1){
        a[++tot1]=num1%2;
        num1/=2;
    }
    while(num2){
        b[++tot2]=num2%2;
        num2/=2;
    }
    return dfs(max(tot1,tot2),1,1,1).second;
}
int main(){
    scanf("%d",&t);
    while(t--){
        scanf("%d %d",&x,&y);
        printf("%lld\n",solve(x,y));
    }
}