2020ICPC上海站C题Sum of Log

2020ICPC上海站C题Sum of Log（数位$dp$）

​ zrr12138

原题地址

首先分析要求的公式可以发现以下几点性质：

1. $j=[i=0]$ 其实是用于保证$i，j$不同时为0
2. $[i\&j=0]$ 表示如果$i,j$二进制下存在某位同时为1则没有贡献
3. $\lfloor lo{g}_2(i+j)+1\rfloor$其实就是$i+j$在二进制下的位数
4. $i,j$交换顺序不改变答案
5. 结合2,3,4条可以知道只有$max(i,j)$最高位的1对答案有贡献

考虑在二进制下对x,y进行数位$dp$(记忆化搜索)：

$dfs(pos,x,y)$ 表示在填第$pos$位（从第0位开始）,x,y表示填当前位之前两个数剩余可用大小，返回值为合法的方案数。

由性质2和性质4可得：所有对于每个位只有3种情况（不妨设第一个数小于等于第二个数）：

• 第一个数填1，第二个数填0
• 第一个数填0，第二个数填1
• 第一个数填0，第二个数填0

对于每个数当前位可以填1的条件是剩余大小大于等于$2^{pos}$

由于x,y大于某个值是剩下的位就可以随便填，这个时候我们将大小缩到这个特定值，以便状态处理。

两个数都可以随便填的状态可以用$dp[pos]$记录

两个数都不可以随便填的状态在状态空间是单一路径，甚至没有。

而一个数可以随便填而另一个数不可以随便填的状态：

• 不可以随便填的数如果当前位可以填1但是填0，之后就转换成了可以随便填的。如果填1那么另一个数只能填0是单条路径。
• 不可以随便填的数只能填0，那么容易想到另外一个数不管填0或1答案都是一样的，不需要分成两条路径。（简单证明：可以随便填的这个数一定大于等于之前提到的特定值，那么不管当前位是否填1，下一位也一定大于等于特定值，压缩之后便成了同一种状态。）

答案记录:

只要枚举最高位1出现的位置进行统计即可，贡献=$\sum$方案数*最高位数。

AC代码

#include <bits/stdc++.h>
#define debug(x) cout<<#x"=" <<x<<'\n';
using namespace std;
const long long mod=1e9+7;
long long dp[40];///
long long  dfs(int pos,int x,int y)
{
    if(x<0||y<0) return 0;
    if(pos==-1) return 1;///递归边界
    int maxx=(1<<(pos+1))-1;///计算特定值
    if(x>y) swap(x,y);///不妨设x<y（性质4）
    if(x>=maxx) x=maxx;
    if(y>=maxx) y=maxx;///压缩到特定值
    if(x==y&&x==maxx&&dp[pos]!=-1) return dp[pos];///如果都可以随便填，返回之前记录的结果
    long long ans=0;
    if(x>=(1<<pos))///第一个数填1
    {
        ans=(ans+dfs(pos-1,x-(1<<pos),y))%mod;
    }
    if(y==maxx)///第二个数为特定值
    {
        ans=(ans+2*dfs(pos-1,x,y-(1<<pos)))%mod;
    }
    else
    {
        if(y>=(1<<pos))///第二个数填1
        {
            ans=(ans+dfs(pos-1,x,y-(1<<pos)))%mod;
        }
        ans=(ans+dfs(pos-1,x,y))%mod;///第二个数填0
    }
    if(x==y&&x==maxx) dp[pos]=ans;///记录都可以随便填的答案
    return ans;
}

int main()
{
    fill(dp,dp+40,-1);
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        long long res=0;
        for(int i=30; i>=0; i--)
        {
            res=(res+(i+1)*dfs(i-1,x-(1<<i),y))%mod;///最高位为第一个数的第i位
            res=(res+(i+1)*dfs(i-1,x,y-(1<<i)))%mod;///最高位为第二个数的第i位
        }
        printf("%lld\n",res);
    }
    return 0;
}