DP——计数类DP、数位统计DP

DP——计数类DP、数位统计DP

• 计数类DP
  • 整数划分
• 数位统计DP
  • 计数问题

---

计数类DP

AcWing 900. 整数划分

• 关键思路：对于x的任意一种划分，该划分的数的集合分为两类，一类包含1，一类不包含1。

#include<iostream>

using namespace std;

const int N = 1100, mod = 1e9 + 7;

int f[N][N], n;

int main()
{
    cin >> n;
    f[0][0] = 1;
    for(int i = 1; i <= n; i++)
        for(int j = i; j <= n; j++)
            f[i][j] = (f[i - 1][j - 1] + f[i][j - i]) % mod;;

    int ans = 0;
    for(int i = 1; i <= n; i++)ans = (ans + f[i][n]) % mod;
    cout << ans;
    return 0;
}

---

---

---

数位统计DP

AcWing 338. 计数问题

• 知识点：分类讨论、一维前缀和
• 思路：（以1 ~ abcdefg中第4位为x的数的个数为例）
  • 前三位 < abc（注：前三位是指固定长度为7，不足在前面补0的前三位，例如hijk的前三位为000）
    • abc * 1000(若x为0，则为(abc-1)*1000
  • 前三位 = abc
    • d > x：1000
    • d = x：efg + 1
    • d < x：0

#include<iostream>

using namespace std;

int P(int x)
{
    int res = 1;
    while(x--)res = res * 10;
    return res;
}

int get_num(int a[], int r, int l)
{
    int res = 0;
    for(int i = r; i >= l; i--)res = res * 10 + a[i];
    return res;
}

int get(int n, int x)
{
    int a[10], cnt = 0;
    while(n)
    {
        a[cnt++] = n % 10;
        n /= 10;
    }
    
    int res = 0;
    for(int i = cnt - 1 - !x; i >= 0; i--)
    {
        if(i < cnt - 1)
        {
            res += get_num(a, cnt - 1, i + 1) * P(i);
            if(!x)res -= P(i);
        }
        if(a[i] > x)res += P(i);
        else if(a[i] == x)res += get_num(a, i - 1, 0) + 1;
    }
    return res;
    
}
int main()
{
    int a, b;
    while(cin >> a >> b, a || b)
    {
        if(a > b)swap(a, b);
        for(int i = 0; i <= 9; i++)
            cout << get(b, i) - get(a - 1, i) << ' ';
        cout << endl;
    }
    return 0;
}