1. 程式人生 > >codeforces 628D 數位dp

codeforces 628D 數位dp

題意是給你兩個長度相等的串,長度為n,求兩串間滿足條件的串的個數。一個串滿足條件當且僅當奇數位不為d,偶數為為d,且串對應的數字可以被m整除。

很典型的數位dp。手殘調了一個小時…老是忘掉取模
思路:dp[i][j][1]表示前i位形成的數字,模m為j,且到達串字首的上限的個數,dp[i][j][0]表示沒有達到字首的上限
然後按位數轉移,討論奇偶的情況。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
using namespace std;
const int
N=2005; const LL mod=1e9+7; char s1[N],s2[N]; int num[N]; LL dp[N][N][2]; int n,m,d; LL f(char *s) { for(int i=1;i<=n;i++)num[i]=s[i]-'0'; memset(dp,0,sizeof(dp)); if(num[1]!=d) dp[1][num[1]%m][1]=1; for(int j=0;j<num[1];j++)if(j!=d) dp[1][j%m][0]++; for(int i=1
;i<=n-1;i++) { for(int j=0;j<m;j++) { if(i%2&&d<num[i+1]) { dp[i+1][(10*j+d)%m][0]+=dp[i][j][1]+dp[i][j][0]; dp[i+1][(10*j+d)%m][0]%=mod; } else if(i%2&&d==num[i+1]) { dp[i+1
][(10*j+d)%m][1]=(dp[i][j][1]+dp[i+1][(10*j+d)%m][1])%mod; dp[i+1][(10*j+d)%m][0]=(dp[i][j][0]+dp[i+1][(10*j+d)%m][0])%mod; } else if(i%2&&d>num[i+1]) dp[i+1][(10*j+d)%m][0]=(dp[i][j][0]+dp[i+1][(10*j+d)%m][0])%mod; else if(i%2==0) { for(int k=0;k<=9;k++)if(k!=d) dp[i+1][(10*j+k)%m][0]=(dp[i+1][(10*j+k)%m][0]+dp[i][j][0])%mod; for(int k=0;k<num[i+1];k++)if(k!=d) dp[i+1][(10*j+k)%m][0]=(dp[i+1][(10*j+k)%m][0]+dp[i][j][1])%mod; if(num[i+1]!=d) dp[i+1][(10*j+num[i+1])%m][1]=(dp[i][j][1]+dp[i+1][(10*j+num[i+1])%m][1])%mod; } } } return dp[n][0][0]+dp[n][0][1]; } int main() { //freopen("a.txt","r",stdin); while(~scanf("%d%d",&m,&d)) { scanf("%s%s",s1+1,s2+1); n=strlen(s1+1); memset(dp,0,sizeof(dp)); LL ans=f(s2)-f(s1); int flag=1,x=0; for(int i=1;i<=n;i++) { if(i%2==0&&num[i]!=d)flag=0; if(i%2&&num[i]==d)flag=0; x=(x*10+num[i])%m; } if(x!=0)flag=0; ans+=flag; printf("%I64d\n",(ans+mod)%mod); } return 0; }