数位dp知识

转自http://blog.csdn.net/zhaoxinfan/article/details/8707605

下面先给出数位DP的背景：

•在给定区间[A,B]内，找满足要求的数。

•要求一般和数大小无关，而与数的组成有关

•例如，递增的，1234, 2579…

•         双峰的，19280,26193…

•         含49的，49, 149, 1492… 

•         整除13的，26, 39…

•麻烦在于，规模大，位数> 100，不能枚举。

•区间往往不是整百整千，边界问题

•

•注意

–记忆化搜索思路清晰

–开适当空间

–寻找合适的状态，简化计算量

为了降低时间复杂度，可以借鉴传统DP中状态转换，打表这些思路，得到了数位DP：

F(A,B) = F(B,0)-F(A-1,0)

暴力+存储 = 记忆化搜索

•暴力：

•暴力枚举每一位(0..9),注意区间边界;与符号的匹配。

 

•dfs(i,j,k,flag)

•枚举第i位的数，匹配str[j],前一位是k，是否达到上限(flag=true,false)

•达到了上限则只能枚举0..num[i]，否则可以枚举0..9

•存储

•dfs(i,j,k,flag)

•设状态与递归参数一致f[i][j][k][flag]，表示当枚举到第i位的数，匹配str[j],前一位是k，是否达到上限(flag=true,false)时，满足要求的数字个数。

•dfs的过程，相当于在填充f，假设f的空间O(100*10*10*2)，则dfs的时间O(20000)

针对上面几种类型的问题，数位DP解决方案如下：（具体可以看http://www.cppblog.com/Yuan/archive/2011/07/15/139299.html）

•整除13

•dfs(i, m, flag)

•枚举第i位数，前面枚举出的数模13的余数m，是否到达上限flag

•

•整除自身各位数CF55D

•dfs(i, m, l, flag)

•枚举第i位数，前面枚举出的数模LCM(0..9)，LCM(前面枚举出的数)，是否达到上限

•

•包含”49”

•dfs(i, k, find, flag)

•枚举第i位数，前一位是k，是否已包含”49”(find)，是否达到上限

•分类讨论：前一位是否为4，当前是否已包含“49”

在这几种类型中，包含49的与微软这道题最为相近，不过要注意的是运算过程中需要把前缀0的情况剔除，最终代码如下：

 #include<iostream>
 #include<algorithm>
 #include<cstdlib>
 #include<cstring>
 using namespace std;
 typedef long long ll;

 #define mem(a,b) memset(a,b,sizeof(a))

 const int L = , P = 1e9+;

 struct RES
 {
     ll all, sum, cnt;
     RES() {}
     RES(int i,int j,int k):all(i),sum(j),cnt(k) {}
 } dp[L];

 ll chkmod(ll x,ll p)
 {
     return (x%p+p)%p;
 }

 int d[L], n;

 RES dfs(int pos, int UP)
 {
     if(pos<)
     {
         return RES(,,);
     }
     if(!UP && ~dp[pos].all)
     {
         return dp[pos];
     }
     RES ret(,,);
     int up=UP?d[pos]:;
     ret.all += dfs(pos-, UP&&up==).all;
     ret.all %= P;
     for(int i=;i<=up;i++)
     {
         int nUP = UP&&i==up;
         for(int j=pos-;j>=-;j--)
         {
             ll tmp = dfs(j, nUP).sum + dfs(j, nUP).cnt * (pos -  - j);
             tmp %= P;
             ret.all += tmp;
             ret.all %= P;
             ret.sum += tmp;
             ret.sum %= P;
             ret.cnt += dfs(j, nUP).cnt;
             ret.cnt %= P;

             nUP = nUP && d[j]==; // !!!
         }
     }

     if(!UP)
     {
         dp[pos] = ret;
     }
     return ret;
 }

 ll cal(ll x)
 {
     n=;
     while(x)
     {
         d[n++]=x%;
         x/=;
     }
     return dfs(n-,).all;
 }

 int main()
 {
     mem(dp,-);
     ll n;
     while(cin>>n)
     {
         cout<<cal(n)<<endl;
     }
     return ;
 }