codeforces 628D 數位dp
阿新 • • 發佈:2019-02-02
題意是給你兩個長度相等的串,長度為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;
}