数位dp解决一类求1~ n n 中求满足某些条件的数有多少,可能给到 10100000 10 100000 左右,所以我们不能用单纯的枚举解决。
于是数位dp给出了这一类问题的解决方案:一位一位地表示状态。
举例:1~ n n 中有多少个数能被3整除(虽然很傻逼,但是我们用数位dp做一下)
表示计算前 i i 位,前位的和模3为 j j 的数有多少个。我们接着枚举下一位,显然可以转移的。但是问题在于我们有一个上界,有可能我们的状态中存在大于的数。
我有两种解决方案(第一种是自己脑补的,然后被卡t了,其实复杂度只是两倍而已):
1.从低位向高位枚举, fvis,i,j f v i s , i , j 表示计算前 i i 位,前位的和模3为 j j 的数有多少个。表示只比较前 i i 位,是否大于。然后加一个高位可以进行转移
2.从高位向低位枚举, fi,j f i , j 表示计算前 i i 位且前位小于等于 n n 的前位,前 i i 位的和模3为的数有多少个。我们可以发现,在这种情况下,如果前 i i 位和不一样,下一位可以枚举0~9,否则只能枚举到0~(n的下一位)。所以我们全部枚举0~9然后把前 i i 位和相同且不合法的情况减掉。(似乎这是一个正确姿势)
然后来做一道例题:A serious math problem
题目大意: f(1234)=1 f ( 1234 ) = 1 ^ 2 2 ^^ 4=4 4 = 4 。 f(x)= f ( x ) = ( x x 所有位上的异或和)。求。
解:
首先很容易想到的就是先求1~ a a 再求1~然后相减就是答案。然后这就是我们的数位dp了。
而且最大的数为9,所以亦或和不会超过16。
fi,j f i , j 表示前 i i 位且不大于亦或和位 j j <script type="math/tex" id="MathJax-Element-117">j</script>的数有多少。
转移似乎比较简单,上述的第一种T了,第二种A了,虽然第二种复杂度也是2e8的,但是它就是过了。
贴个代码:
include
#include<cstdio>
#include<cstring>
using namespace std;
int mod=1000000007;
int f[100005][16];
int T,n;
char a[100005],b[100005];
int solv(char* x)
{
int t,ret=0,pre=0;
int len=strlen(x+1);
memset(f,0,sizeof(f));
f[0][0]=1;
for(register int i=1;i<=len;i++)
{
t=x[i]-'0';
for(int w=0;w<16;w++)
for(int m=0;m<=9;m++)
f[i][w^m]=(f[i][w^m]+f[i-1][w])%mod;
for(int m=t+1;m<=9;m++)
f[i][pre^m]--;
pre^=t;
}
for(register int w=0;w<=15;w++)
ret=(ret+(long long)f[len][w]*w)%mod;
return ret;
}
int main()
{
scanf("%d",&T);n=T;
while(n--)
{
int p=0;
scanf("%s%s",a+1,b+1);
int len=strlen(a+1);
for(register int i=1;i<=len;i++)
p=p^(a[i]-'0');
p=(solv(b)-solv(a)+p)%mod;
if(p<0) p+=mod;
printf("Case #%d: %d\n",T-n,p);
}
}