简单的数位dp练习

【BZOJ1833】【ZJOI2010】数字计数
Description
给定两个正整数a和b，求在[a,b]中的所有整数中，每个数码(digit)各出现了多少次。
Input
输入文件中仅包含一行两个整数a、b，含义如上所述。
Output
输出文件中包含一行10个整数，分别表示0-9在[a,b]中出现了多少次。
Sample Input
1 99
Sample Output
9 20 20 20 20 20 20 20 20 20

#include<cstdio>#include<algorithm>using namespace std;#define ll long longll a, b, num[15], f[2][15], p[15];void init(ll x, ll *n){    ll k=0;    for(ll i=x; i!=0; i/=10) num[++k] = i%10;    for(ll i=k; i>=1; i--)    {        if(i!=1) for(ll j=0; j<=9; j++) n[j] += (i-1)*p[i-1]*num[i]; //i位数前i-1位组合         for(ll j=0; j<num[i]; j++) n[j] += p[i]; //i位数第i位 小于最高位数字         n[num[i]] += x%p[i]+1; //最高位      }    for(ll i=k; i>=2; i--) n[0] -= (k-i+1) * p[i-1] * 9;    n[0] -= k;}int main(){//  freopen("input.txt", "r", stdin);    scanf("%I64d%I64d", &a, &b); p[1] = 1;    for(ll i=2; i<=12; i++) p[i] = 10*p[i-1];    init(a-1, f[0]); init(b, f[1]);    for(ll i=0; i<=9; i++) printf("%I64d ", f[1][i]-f[0][i]);    return 0;}

---

【BZOJ1026】【SCOI2009】windy数
Description
windy定义了一种windy数。不含前导零且相邻两个数字之差至少为22的正整数被称为windy数。 windy想知道，在A和B之间，包括AA和BB，总共有多少个windy数？
Input
包含两个整数，A,BA,B。
Output
一个整数
Sample Input
1 10
Sample Output
9
HINT
100%的数据，满足 1≤A≤B≤2000000000

#include<cstdio>#include<algorithm>using namespace std;int A, B;int num[20], f[20][20];int calc(int x){    int t=0, ans=0;    while(x!=0) {num[++t] = x % 10; x /= 10;}     num[t+1] = 0;    for(int i=1; i<t; i++)        for(int j=1; j<=9; j++) ans += f[i][j];    for(int i=1; i<num[t]; i++) ans += f[t][i];    for(int i=t-1; i>=1; i--)    {        for(int j=0; j<num[i]; j++)            if(abs(j-num[i+1])>=2) ans += f[i][j];        if(abs(num[i]-num[i+1])<2) break;    }    return ans;}int main(){    scanf("%d%d", &A, &B);     for(int i=0; i<=9; i++) f[1][i] = 1;    for(int i=2; i<=10; i++)        for(int j=0; j<=9; j++)            for(int k=0; k<=9; k++)                if(abs(j-k)>=2) f[i][j] += f[i-1][k];    printf("%d", calc(B+1)-calc(A));    return 0;}