DP - 数位DP -数字游戏(Ⅰ+Ⅱ)

DP - 数位DP -数字游戏(Ⅰ+Ⅱ)

---

1、数字游戏Ⅰ

科协里最近很流行数字游戏。

某人命名了一种不降数，这种数字必须满足从左到右各位数字呈非下降关系，如 123，446。

现在大家决定玩一个游戏，指定一个整数闭区间 [a,b]，问这个区间内有多少个不降数。

输入格式
输入包含多组测试数据。

每组数据占一行，包含两个整数 a 和 b。

输出格式
每行给出一组测试数据的答案，即 [a,b] 之间有多少不降数。

数据范围
1≤a≤b≤2³¹−1

输入样例：
1 9
1 19
输出样例：
9
18

---

分析：

$dp函数求1$~$N中不降数的个数。$

$首先将N每一位存入数组V=a_{n-1}{a}_{n-2}...a_0，$

$从高位到低位[n-1,0]依次遍历：$

$设f[i,j]表示长度为i，且最高位为j的数内，不降数的个数。$
$则f[i,j]+=f[i-1,k]，其中k>=j。这里得到f数组的转移方程。$
$last表示上一位数的大小。$

$考虑第i位数a_i$

$①、若last<=a_i<V_i，则无论a_i{a}_{i-1}...a_0取何值，均不会超过V，总方案数res+=f[i+1][j]。$

$②、若last>V_i，此时无论如何都无法选出合法方案，直接退出循环，否则更新last=V_i。$

$③、若i=0，即考虑到最后一位a_0，还要加上V本身这个方案。$

代码：

#include<iostream>
#include<vector>

using namespace std;

const int N=15;

int f[N][N];

void cal()
{
    for(int i=0;i<=9;i++) f[1][i]=1;
    
    for(int i=2;i<N;i++)
        for(int j=0;j<=9;j++)
            for(int k=j;k<=9;k++)
                f[i][j]+=f[i-1][k];
}

int dp(int n)
{
    if(!n) return 1;
    vector<int> V;
    while(n) V.push_back(n%10),n/=10;
    
    int res=0,last=0;
    for(int i=V.size()-1;i>=0;i--)
    {
        int x=V[i];
        for(int j=last;j<x;j++)
            res+=f[i+1][j];
            
        if(x<last) break;
        last=x;
        
        if(!i) res++;
    }
    
    return res;
}

int main()
{
    cal();
    
    int l,r;
    while(cin>>l>>r) cout<<dp(r)-dp(l-1)<<endl;
    
    return 0;
}

---

2、数字游戏Ⅱ

由于科协里最近真的很流行数字游戏。

某人又命名了一种取模数，这种数字必须满足各位数字之和 mod P 为 0。

现在大家又要玩游戏了，指定一个整数闭区间 [a.b]，问这个区间内有多少个取模数。

输入格式
输入包含多组测试数据，每组数据占一行。

每组数据包含三个整数 a,b,P。

输出格式
对于每个测试数据输出一行结果，表示区间内各位数字和 mod P 为 0 的数的个数。

数据范围
1≤a,b≤2³¹−1,
1≤P<100

输入样例：
1 19 9
输出样例：
2

---

分析:

$与第一题类似的，$

$把N的每一位存入数组，$

$从高位到低位依次遍历每一位数，$

$设f[i,j,k]表示长度为i，末尾数字为j，且各位数字之和对P取模的余数是k的合法方案总数。$

$形如：$

$由(j+x+S)\%P=k得(x+S)\%P=(k-j)\%P，则f[i][j][k]可由f[i-1][x][(k-j)\%P]转移而来。$

$即f[i][j][k]+=f[i-1][x][(k-j)\%P]，求f数组的转移方程。$

注意：$负数取模。$

$设last=V_{n-1}+V_{n-2}+...+V_{i+1}，第i位数填j$

$$\begin{gathered} ①、若j<V_i，无论S_a=a_i{a}_{i-1}...a_0这i+1位填什么数，都不会超过V。 \\ 又因为需满足总和(S_a+last)\%P=0，故S_a对P取模的余数应当是(-last)\%P。 \\ 总方案数res+=f[i+1][j][(-last)\%P]。 \end{gathered}$$

$②、更新last，若枚举到最后一位a_0，且总和last\%P=0，还要加上V本身这一个方案。$

代码：

#include<iostream>
#include<vector>
#include<cstring>

using namespace std;

const int N = 11;

int f[N][N][110];
int l,r,P;

int mod(int x,int p)
{
    return (x%p+p)%p;
}

void cal()
{
    memset(f,0,sizeof f);
    
    for(int i=0;i<=9;i++) f[1][i][i%P]++;
    
    for(int i=2;i<N;i++)
        for(int j=0;j<=9;j++)
            for(int k=0;k<P;k++)
                for(int x=0;x<=9;x++)
                    f[i][j][k]+=f[i-1][x][mod(k-j,P)];
}

int dp(int n)
{
    if(!n) return 1;
    
    vector<int> V;
    while(n) V.push_back(n%10),n/=10;
    
    int res=0,last=0;
    for(int i=V.size()-1;i>=0;i--)
    {
        int x=V[i];
        for(int j=0;j<x;j++)
            res+=f[i+1][j][mod(-last,P)];
        
        last+=x;
        
        if(!i && last%P==0) res++;
    }
    
    return res;
}

int main()
{
    while(cin>>l>>r>>P)
    {
        cal();
        cout<<dp(r)-dp(l-1)<<endl;
    }
    
    return 0;
}