4037: [HAOI2015]数字串拆分

4037: [HAOI2015]数字串拆分

Time Limit: 10 Sec   Memory Limit: 256 MB
Submit: 255   Solved: 156
[ Submit][ Status][ Discuss]

Description

你有一个长度为n的数字串。定义f(S)为将S拆分成若干个1~m的数的和的方案数，比如m=2时，f(4)=5，分别为4=1+

1+1+1你可以将这个数字串分割成若干个数字（允许前导0），将他们加起来，求f，并求和。比如g(123)=f(1+2+3)

+f(1+23)+f(12+3)+f(123)。已知字符串和m后求答案对998244353（7×17×223+1，一个质数）取模后的值。

Input

第一行输入一个字符串，第二行输入m

Output

仅输出一个数表示答案

Sample Input

123
3

Sample Output

394608467

HINT

 对于100%的数据，字符串长度不超过500，m<=5

Source

鸣谢bhiaibogf提供

[ Submit][ Status][ Discuss]



先考虑f(n)的求法，显然f(n) = f(n-1) + f(n-2) + ... + f(n-m)，也就是一个m阶递推数列，f[0] = 1

这个东西显然是能用矩阵来维护的，反正写一个m * m的转移矩阵。。。。

不妨记f(n)最终使用的转移矩阵是h(n)

那么f(a1 + a2 + ... + ak)使用的转移矩阵就能写成h(a1) * h(a2) * ... * h(ak)

通过矩阵乘法满足结合律推出，这是一个很不错的性质

记D[i][j]为原数字串中i ~ j位构成的数字的转移矩阵，

g[i]为以i结尾之前所有种类转移矩阵的和，g[i] = ∑g[j] * D[j + 1][i]

因为矩阵乘法也满足分配率，所以这样的方程是对的

最后拿g[n]瞎算一下就得出答案了

O(5 * n^2 * m^3)，这个常数5是预处理D数组的时候自带的。。。可能有更优的吧，反正我是这样了。。

这样子复杂度挺大的，所以常数得注意点

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
 
const int N = 505;
const int M = 5;
typedef long long LL;
const LL mo = 998244353;
 
int n,m;
char s[N];
 
struct data{
    LL a[M][M];
    data operator * (const data &B)
    {
        data c; memset(c.a,0,sizeof(c.a));
        for (int k = 0; k < m; k++)
            for (int i = 0; i < m; i++)
                for (int j = 0; j < m; j++)
                    c.a[i][j] += a[i][k] * B.a[k][j];
        for (int i = 0; i < m; i++)
            for (int j = 0; j < m; j++)
                c.a[i][j] %= mo;
        return c;
    }
    data operator += (const data &B)
    {
        for (int i = 0; i < m; i++)
            for (int j = 0; j < m; j++)
            {
                a[i][j] += B.a[i][j];
                if (a[i][j] >= mo) a[i][j] -= mo;
            }
    }
}D[N][N],mi[10],g[N];
 
data ksm(data &k)
{
    k = k * k; data ret = k;
    k = k * k; k = k * k; ret = ret * k;
    return ret;
}
 
int main()
{
    #ifdef DMC
        freopen("DMC.txt","r",stdin);
    #endif
     
    scanf("%s",s + 1); cin >> m;
    n = strlen(s + 1); for (int i = 1; i <= n; i++) s[i] -= '0';
    for (int i = 1; i < m; i++) mi[1].a[i][i - 1] = 1;
    for (int i = 0; i < m; i++) mi[1].a[i][m - 1] = g[0].a[i][i] = 1;
    for (int i = 2; i < 10; i++) mi[i] = mi[i - 1] * mi[1];
    for (int i = 1; i <= n; i++)
    {
        data now; memset(now.a,0,sizeof(now.a));
        for (int j = 0; j < m; j++) now.a[j][j] = 1;
        for (int j = i; j <= n; j++)
        {
            now = ksm(now);
            if (s[j] > 0) now = now * mi[s[j]];
            D[i][j] = now;
        }
    }
    for (int i = 1; i <= n; i++)
        for (int j = i - 1; j >= 0; j--)
            g[i] += g[j] * D[j + 1][i];
    LL Ans = 0;
    for (int i = 0; i < m; i++) Ans += g[n].a[0][i];
    cout << Ans % mo << endl;
    return 0;
}