2020ICPC上海站C题Sum of Log(数位 d p dp dp)
zrr12138
原题地址
首先分析要求的公式可以发现以下几点性质:
- j = [ i = 0 ] j=[i=0] j=[i=0] 其实是用于保证 i , j i,j i,j不同时为0
- [ i & j = 0 ] [i\&j=0] [i&j=0] 表示如果 i , j i,j i,j二进制下存在某位同时为1则没有贡献
- ⌊ l o g 2 ( i + j ) + 1 ⌋ \lfloor log_2(i+j)+1 \rfloor ⌊log2(i+j)+1⌋其实就是 i + j i+j i+j在二进制下的位数
- i , j i,j i,j交换顺序不改变答案
- 结合2,3,4条可以知道只有 m a x ( i , j ) max(i,j) max(i,j)最高位的1对答案有贡献
考虑在二进制下对x,y进行数位 d p dp dp(记忆化搜索):
d f s ( p o s , x , y ) dfs(pos,x,y) dfs(pos,x,y) 表示在填第 p o s pos pos位(从第0位开始),x,y表示填当前位之前两个数剩余可用大小,返回值为合法的方案数。
由性质2和性质4可得:所有对于每个位只有3种情况(不妨设第一个数小于等于第二个数):
- 第一个数填1,第二个数填0
- 第一个数填0,第二个数填1
- 第一个数填0,第二个数填0
对于每个数当前位可以填1的条件是剩余大小大于等于 2 p o s 2^{pos} 2pos
由于x,y大于某个值是剩下的位就可以随便填,这个时候我们将大小缩到这个特定值,以便状态处理。
两个数都可以随便填的状态可以用 d p [ p o s ] dp[pos] 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;
}