简单的数位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 long

ll 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;
}