hdu 3967 Zero's Number (不允许前导0(新写法)的数位dp)
题意:给出l,r,k,定义f(n,k)为将数n分成左右两个非空的部分,再求和之后能被k整除的方案数。
现在问区间[l,r]中所有f(i,k)的和。
思路:数位dp...
枚举一下分点即可。。想到这个这题就A了。。。
然后相当于做分点个数个数位dp...求和即可。
dp[i][j][k][p]表示第i位,前半部分%k的结果,后半部分%k的结果,是否有前导0.
然后关于前导0这个。。。换了一种写法。。。加了一个状态在dp数组里。。。
不然每次都要一个if。。。感觉有点丑。。。。这样写简介了一点。。。
以及。。因为每次k是不同的。。。我dp状态记录的时候又没有记录k...所以记得每次初始化成-1。。。
因为忘记这个结果第二个样例调了好久一直是31。。。。
/* ***********************************************
Author :111qqz
Created Time :Thu 29 Sep 2016 02:49:50 PM CST
File Name :code/hdu/3967.cpp
************************************************ */
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <cmath>
#include <cstdlib>
#include <ctime>
#define fst first
#define sec second
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define ms(a,x) memset(a,x,sizeof(a))
typedef long long LL;
#define pi pair < int ,int >
#define MP make_pair
using namespace std;
const double eps = 1E-8;
const int dx4[4]={1,0,0,-1};
const int dy4[4]={0,-1,1,0};
const int inf = 0x3f3f3f3f;
int k;
LL l,r;
int digit[20];
LL dp[20][20][20][20][2];
LL dfs( int pos,LL sum1,LL sum2,bool limit,int cut,int prehasnonzero)
{
if (pos==0) return (sum1+sum2)%k==0;
if (!limit&&dp[pos][sum1%k][sum2%k][cut][prehasnonzero]!=-1) return dp[pos][sum1%k][sum2%k][cut][prehasnonzero];
int mx = limit?digit[pos]:9;
LL res = 0 ;
for ( int i = 0 ; i <= mx; i++)
{
if (pos<cut) res = res + dfs(pos-1,sum1,sum2*10+i,limit&&i==mx,cut,prehasnonzero||i!=0);
else
{
if (!prehasnonzero&&pos==cut&&i==0) continue; //前半部分可以有前导0,但是不能全为0.
res = res + dfs(pos-1,sum1*10+i,sum2,limit&&i==mx,cut,prehasnonzero||i!=0);
}
}
if (!limit) dp[pos][sum1%k][sum2%k][cut][prehasnonzero] = res;
return res;
}
LL solve( LL n)
{
// if (n<10) return 0;
ms(digit,0);
int len = 0 ;
ms(dp,-1);// 每次计算的时候都要初始化dp。。。因为d不一样。。。。。。
while (n)
{
digit[++len] = n % 10;
n /= 10;
}
LL res = 0 ;
for ( int i = 2 ; i <= len ; i++)
res = res + dfs(len,0,0,true,i,0);
return res;
}
int main()
{
#ifndef ONLINE_JUDGE
freopen("code/in.txt","r",stdin);
#endif
while (~scanf("%lld%lld%d",&l,&r,&k))
{
LL ans = solve(r) - solve(l-1);
printf("%lld\n",ans);
}
#ifndef ONLINE_JUDGE
fclose(stdin);
#endif
return 0;
}