#1560 : H国的身份证号码II(dp+矩阵快速幂)

#1560 : H国的身份证号码II

 

时间限制:10000ms

单点时限:1000ms

内存限制:256MB

描述

H国的身份证号码是一个N位的正整数(首位不能是0)。此外，由于防伪需要，一个N位正整数是合法的身份证号码当且仅当每位数字都小于等于K，并且任意相邻两位数字的乘积也小于等于K。

例如对于K=5, 101、211、210等都是合法的号码，而106、123、421等都是非法的号码。

给定一个正整数N以及K，H国总统想知道一共有多少个合法的号码可用。

输入

两个整数N和K。

对于30%的数据，1 ≤ N ≤ 10  

对于50%的数据，1 ≤ N ≤ 1000000

对于100%的数据，1 ≤ N ≤ 1012，1 ≤ K ≤ 81。

输出

合法号码的总数。由于答案可能非常大，你只需要输出答案对10⁹+7取模的结果。

样例输入

2 4

样例输出

12

//dp[i][j] 代表 i 长度，结尾为 j 的合法方案数

那么容易想到

i = 1 : dp[1][j] = 1 (j<=k)

i > 1 : dp[i][j] = ∑dp[i-1][x] (x<=k&&j<=k&&j*x<=k)

可以发现，可以用矩阵快速幂优化，O(10³*logn)

# include <cstdio>
# include <cstring>
# include <cstdlib>
# include <iostream>
# include <vector>
# include <queue>
# include <stack>
# include <map>
# include <bitset>
# include <sstream>
# include <set>
# include <cmath>
# include <algorithm>
# pragma  comment(linker,"/STACK:102400000,102400000")
using namespace std;
# define LL          long long
# define pr          pair
# define mkp         make_pair
# define lowbit(x)   ((x)&(-x))
# define PI          acos(-1.0)
# define INF         0x3f3f3f3f3f3f3f3f
# define eps         1e-8
# define MOD         1000000007

inline int scan() {
    int x=0,f=1; char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-') f=-1; ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0'; ch=getchar();}
    return x*f;
}
inline void Out(int a) {
    if(a<0) {putchar('-'); a=-a;}
    if(a>=10) Out(a/10);
    putchar(a%10+'0');
}
#define MX 10
/**************************/
struct Mat
{
    LL m[MX][MX];
}unit,di,fir;

LL n,k;

Mat mult(Mat a,Mat b)
{
    Mat c;
    for (int i=0;i<MX;i++)
        for (int j=0;j<MX;j++)
        {
            c.m[i][j]=0;
            for (int k=0;k<MX;k++)
                c.m[i][j]=(c.m[i][j]+a.m[i][k]*b.m[k][j])%MOD;
        }
    return c;
}

Mat cal(LL p)
{
    Mat ret = unit, b = di;
    while (p)
    {
        if (p&1) ret = mult(ret,b);
        b = mult(b,b);
        p/=2;
    }
    return ret;
}

void Init()
{
    for (int i=0;i<MX;i++)
        unit.m[i][i]=1;
    for (int i=1;i<MX;i++)
    {
        if (i<=k)
            fir.m[i][0]=1;
    }
    for (int i=0;i<MX;i++)
    {
        for (int j=0;j<MX;j++)
        {
            if (i<=k&&j<=k&&i*j<=k)
                di.m[i][j]=di.m[j][i]=1;
        }
    }
}

int main()
{
    scanf("%lld%lld",&n,&k);
    Init();
    Mat sa = fir;
    Mat sb = cal(n-1);
    sb = mult(sb,sa);

    LL ans = 0;
    for (int i=0;i<MX;i++)
    {
        ans = (ans + sb.m[i][0])%MOD;
    }
    printf("%lld
",ans);
    return 0;
}