[51nod 1197] 字符串的数量 V2

Description

用N个不同的字符（编号1 - N），组成一个字符串，有如下要求：
(1) 对于编号为i的字符，如果2 * i > n，则该字符可以作为结尾字符。如果不作为结尾字符而是中间的字符，则该字符后面可以接任意字符。
(2) 对于编号为i的字符，如果2 * i <= n，则该字符不可以作为结尾字符。作为中间字符，那么后面接的字符编号一定要 >= 2 * i。
问有多少长度为M且符合条件的字符串，由于数据很大，只需要输出该数Mod 10^9 + 7的结果。
例如：N = 2，M = 3。则abb, bab, bbb是符合条件的字符串，剩下的均为不符合条件的字符串。

Input

输入2个数，N, M中间用空格分割，N为不同字符的数量，M为字符串的长度。(2 <= N <= 10^6, 2 <= M <= 10^18)

Output

输出符合条件的字符串的数量。由于数据很大，只需要输出该数Mod 10^9+7的结果。

分析

n很大，以它为状态一定会爆。但是我们发现：对于一个数i（i*2>n），它如果不是结尾，下一个大于n的数和它的距离不超过20（因为对于一个i（i *2≤n），下一个数至少是它的两倍，$\log_{2}^n$小于20）。
所以可以先预处理出来一个数组sum[x]表示两个i（i*2>n）之间有x个数，可能的情况有多少个。设f[i][j]为当前一个序列最后一个元素为i（i∈[1,n]），长度为j，且第1到i-1个数都满足x*2≤n，方案数有多少。转移：$f[i][j]=\sum_{k=0}^\frac{i}{2} f[k][j-1]$ 边界条件为f[0][0]=1 打个前缀和即可在n*20的时间内预处理。然后可以相应的求出v数组。

接着设s[i]为长度为i的满足条件的序列有多少个。
$s[i]=\sum_{j=0}^{20} s[i-j]*v[j]$

然后矩阵乘法解决。时间复杂度$O(20^3log(m))$

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int mo=1e9+7,maxn=1000005;

typedef long long LL;

int n,f[22][22],a[22][22],b[22][22],s[maxn][22],v[maxn];

LL m;

void mulf()
{
    memset(b,0,sizeof(b));
    for (int i=0;i<20;i++)
    {
        for (int j=0;j<20;j++)
        {
            for (int k=0;k<20;k++)
            {
                b[i][j]=(b[i][j]+(LL)f[i][k]*f[k][j]) % mo;
            }
        }
    }
    memcpy(f,b,sizeof(b));
}

void mula()
{
    memset(b,0,sizeof(b));
    for (int i=0;i<20;i++)
    {
        for (int j=0;j<20;j++)
        {
            for (int k=0;k<20;k++)
            {
                b[i][j]=(b[i][j]+(LL)f[i][k]*a[k][j]) % mo;
            }
        }
    }
    memcpy(f,b,sizeof(b));
}

void quick(LL x)
{
    if (!x) return;
    quick(x/2);
    mulf();
    if (x % 2==1) mula();
}

int main()
{
    scanf("%d%lld",&n,&m);
    s[0][0]=1;
    for (int i=1;i<=n;i++)
    {
        for (int j=1;j<=20;j++)
            s[i][j]=(s[i-1][j]+s[i/2][j-1]) % mo;
        s[i][0]=1;
    }
    for (int i=1;i<=20;i++) v[i]=(s[n][i]-s[n/2][i]+mo) % mo;
    for (int i=0;i<20;i++)
    {
        if (i) a[i-1][i]=1;
        a[i][0]=v[i+1];
        f[i][i]=1;
    }
    quick(m);
    printf("%d\n",f[0][0]);
    return 0;
}