• [bzoj1833][ZJOI2010]count 数字计数——数位dp


    题目:

    (传送门)[http://www.lydsy.com/JudgeOnline/problem.php?id=1833]

    题解:

    第一次接触数位dp,真的是恶心。
    首先翻阅了很多很多一维dp,因为要处理前缀0,所以根本搞不懂。
    查询了dalaolidaxin的博客,又查阅了资料:
    初探数位dp
    才完全弄懂这个题。
    具体的,我们设
    f[i][j][k]为考虑所有i位数,最高位为j数,之中k的数目。
    我们可以得出方程:

    [f[i][j][k] = sum f[i-1][l][k] (j!=k) ]

    [f[i][j][k] = sum f[i-1][l][k] + 10^{i-1} (j==k) ]

    我们对这个方程作出解释:
    前一项非常好理解,后一项的话就是前(i-1)位数共有(10^{i-1})个,对于其中每一个,我们都可以在前面加k。
    这样我们预处理出来了f。
    然后我们考虑对于n分块计算。
    以n = 4321为例。
    首先统计3位及以下的数,这些数字没有限制,直接加就好。
    然后统计4位数。
    对于一个4位数,我们一位一位向下考虑,如果最高位<k,直接加,如果=k,加上n+1
    具体见代码。

    代码

    #include <cstdio>
    #include <cstring>
    using namespace std;
    #define ll long long
    const int N = 25;
    struct node {
      ll a[N];
      node() { memset(a, 0, sizeof(a)); }
      ll &operator[](const int &x) { return a[x]; }
    };
    node operator+(const node &x, const node &y) {
      node tmp;
      for (int i = 0; i <= 9; i++)
        tmp.a[i] = x.a[i] + y.a[i];
      return tmp;
    }
    int len, a[N];
    ll pow[N];
    node f[N][N];
    void init(ll n) {
      len = 0;
      while (n) {
        a[++len] = n % 10;
        n /= 10;
      }
      for (int i = 0; i <= 9; i++)
        f[1][i][i] = 1;
      for (int i = 2; i <= 14; i++) {
        for (int j = 0; j <= 9; j++) {
          for (int k = 0; k <= 9; k++)
            f[i][j] = f[i][j] + f[i - 1][k];
          f[i][j][j] += pow[i - 1];
        }
      }
    }
    node calc(ll n) {
      node ans;
      if (!n)
        return ans;
      memset(f, 0, sizeof(f));
      init(n);
      //统计前len-1位
      for (int i = 1; i <= len - 1; i++) {
        for (int j = 1; j <= 9; j++) {
          ans = ans + f[i][j];
        }
      }
      //开始统计len位数
      for (int i = 1; i <= a[len] - 1; i++)
        ans = ans + f[len][i];
      n %= pow[len - 1];
      ans[a[len]] += n + 1; //对于每一个最高位都可以统计一发
      for (int i = len - 1; i; i--) {
        for (int j = 0; j < a[i]; j++)
          ans = ans + f[i][j];
        n %= pow[i - 1];
        ans[a[i]] += n + 1;
      }
      return ans;
    }
    int main() {
      pow[0] = 1;
      for (int i = 1; i <= 14; i++)
        pow[i] = pow[i - 1] * 10;
      ll x, y;
      scanf("%lld %lld", &x, &y);
      node ans1 = calc(y), ans2 = calc(x - 1);
      for (int i = 0; i <= 8; i++)
        printf("%lld ", ans1[i] - ans2[i]);
      printf("%lld
    ", ans1[9] - ans2[9]);
      return 0;
    }
    
  • 相关阅读:
    62. Unique Paths
    24. Swap Nodes in Pairs
    83. Remove Duplicates from Sorted List
    21. Merge Two Sorted Lists
    141. Linked List Cycle
    268. Missing Number
    191. Number of 1 Bits
    231. Power of Two
    9. Palindrome Number
    88. Merge Sorted Array
  • 原文地址:https://www.cnblogs.com/gengchen/p/6407547.html
Copyright © 2020-2023  润新知