fzu 1896 神奇的魔法数字(数位dp)

数位dp就不讲了,网上一搜一大把(才不是因为我懒呢)。

依旧是状态的定义。。。。
dp[i][j]表示前i位最高位是j的神奇的魔法数的个数。
cnt一个a,b+1的神奇的魔法数的个数就对了。(左闭右开)
这里是代码:

#include
#include
#include
#include
#include
#include
using namespace std;
int dp[15][15];
int m;
void init()
{
    memset(dp,0,sizeof(dp));
    for(int i = 0; i<=10; i++)
    {
        dp[1][i] = 1;
    }
    for (int i = 2; i < 12; i++)
        for (int j = 0; j < 10; j++)
            for (int k = 0; k < 10; k++)
                if (abs(j-k) >= m)
                    dp[i][j] += dp[i-1][k];
}
int solve(int n)
{
    int b[15];
    b[0]=0;
    int dit = 0;
    while(n)
    {
        b[++dit]=n%10;
        n/=10;
    }
    int ans = 0;
    for(int i = 1; ifor(int j = 1; j<10; j++)
        {
            ans+=dp[i][j];
        }
    }
    for (int i = 1; i < b[dit]; i++)
        ans += dp[dit][i];
    for (int i = dit-1; i > 0; i--)
    {
        for (int j = 0; j < b[i]; j++)
            if (abs(j-b[i+1]) >= m)
                ans += dp[i][j];
        if (abs(b[i]-b[i+1]) < m)
            break;
    }
    return ans;
}
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        int a,b;
        cin>>a>>b>>m;
        init();
        cout<1)-solve(a)<

你可能感兴趣的:(「ac妹」dp)