[hdu5564] Clarke and digits

问题描述

克拉克是一名人格分裂患者。某一天，克拉克变成了一个研究人员，在研究数字。
他想知道在所有长度在$[l, r]$之间的能被$7$整除且相邻数位之和不为$k$的正整数有多少个。

输入描述

第一行一个整数$T (1 ≤ T ≤ 5)$，表示数据的组数。
每组数据只有一行三个整数$l, r, k(1 ≤ l ≤ r ≤ 10 , 0 ≤ k ≤ 18)$。

输出描述

每组数据输出一行一个数，表示答案。由于答案太大，你只需对$10^9 + 7$取模即可。

输入样例

2
1 2 5
2 3 5

输出样例

13
125

Hint

第一个样例有$13$个数满足，分别是：$7, 21, 28, 35, 42, 49, 56, 63, 70, 77, 84, 91, 98$。

题解

这题真是妙啊，做了才发现矩阵快速幂dp的正确姿势，不过这两天智商不在线，一个小小的取模问题就调了两个小时，看来要多睡觉啊。
首先我们可以推出dp方程（这里就懒得写了）。
然后说一下构造转移矩阵的一般方法：假设$dp[s1]+=k*dp[s2]$，那么转移矩阵$B[s1][s2]+=k$，其实很好理解，想一想就懂了。
另外，不要以为$a*b^{r}-a*b^{l-1}$就是答案，因为这两项都不是前缀和，所以要给转移矩阵加一列记录前缀和，具体处理可以参考其他博客。

Code

#include<bits/stdc++.h>
using namespace std;

const long long mod=1e9+7;
const int n=70;
long long t,l,r,x;
struct matrix{
    long long m[72][72];
    matrix operator * (const matrix &b)const{
        matrix c;
        for(int i=0;i<=n;i++)
          for(int j=0;j<=n;j++)
            c.m[i][j]=0;
        for(int i=0;i<=n;i++)
          for(int j=0;j<=n;j++)
            for(int k=0;k<=n;k++)
              c.m[i][j]=(c.m[i][j]+m[i][k]*b.m[k][j])%mod;
        return c;
    }
}z,a;

matrix fpow(matrix b,long long w){
    matrix c;
    for(int i=0;i<=n;i++)
      for(int j=0;j<=n;j++){
        if(i==j)c.m[i][j]=1;
        else c.m[i][j]=0;
      }
    while(w){
        if(w&1)c=c*b;
        b=b*b;
        w>>=1;
    }
    return c;
}

long long solve(long long w){
    if(!w)return 0;
    for(int i=0;i<=n;i++)
      for(int j=0;j<=n;j++)
        z.m[i][j]=a.m[i][j]=0;
    for(int i=1;i<=9;i++)a.m[0][(i%7)*10+i]=1;
    for(int i=0;i<7;i++)
      for(int j=0;j<=9;j++)
        for(int p=0;p<=9;p++)
          if(j+p!=x)
            z.m[i*10+j][((i*10+p)%7)*10+p]=1;
    z.m[70][70]=1;
    for(int i=0;i<=9;i++)z.m[i][70]=1;
    z=fpow(z,w);a=a*z;
    return a.m[0][70];
}

int main()
{
    scanf("%lld",&t);
    while(t--){
        scanf("%lld%lld%lld",&l,&r,&x);
        printf("%lld\n",((solve(r)-solve(l-1))%mod+mod)%mod);
    }
    return 0;
}