2014年北京邀请赛 E题 矩阵+dp

题目大意：判断有多少个长度为n的字符串，满足该字符串的任意一个子串不含有0,1,2,3,4……k的任何一个排列。

思路：f[i][j],表示长度为i 的字符串末尾有j个不同的数字。

           f[n+1][1]=f[n][1]+f[n][2]+f[n][3]+……+f[n][k];

           f[n+1][2]=k*f[n][1]+f[n][2]+f[n][3]+……+f[n][k];

           f[n+1][3]=0*f[n][1]+(k-1)*f[n][2]+f[n][3]+……+f[n][k];

           f[n+1][4]=0*f[n][1]+0*f[n][2]+(k-2)*f[n][3]+……+f[n][k];

            .............

           f[n+1][k]=0*f[n][1]+0*f[n][2]+0*f[n][3]+……+（k+1-(k-1)）*f[n][k-1]+f[n][k];

         系数矩阵：1,      1,        1      …… 1，1

                             k,      1,        1     …… 1， 1

                             0,     k-1,      1     …… 1，1

                             0,      0，   k-2    …… 1，1

                               …………

                            0,       0，     0     ……   2,   1 

  矩阵转移方程： f[n][1]                 1,      1,        1      …… 1，1              （ n-1）                    f[1][1]

                              f[n][2]                  k,      1,        1     …… 1， 1                                                f[1][2]     

                              f[n][3]       =        0,     k-1,      1     …… 1，1                                     *           f[1][3]  

                               ……                   ……                                                                                      ……

                              f[n][k]                 0,       0，     0     ……   2,   1                                               f[1][k]

其中： f[1][1]=k+1，f[1][2] =f[1][3]=……=f[1][k]=0;    

最终结果为：f[n+1][1]+f[n+1][2]+……+ f[n+1][k].

 详见代码：

#include <stdio.h>
#include <string.h>
#include <iostream>
using namespace std;
typedef long long LL;
const int MOD=20140518;
struct Matrix
{
    LL m[10][10];
};
Matrix I;
LL n,k;
void Input()
{
    for(int i=0;i<k;i++)
        for(int j=0;j<k;j++)
             I.m[i][j]=(i==j);
}

Matrix multi(Matrix a,Matrix b)
{
    Matrix c;
    for(int i=0;i<k;i++)
        for(int j=0;j<k;j++)
        {
            c.m[i][j]=0;
            for(int kk=0;kk<k;kk++)
            {
                c.m[i][j]+=a.m[i][kk]*b.m[kk][j]%MOD;
            }
            c.m[i][j]%=MOD;
        }
    return c;
}
Matrix quick_mod(Matrix a,LL k)
{
    Matrix ans=I;
    while(k!=0)
    {
        if(k&1)
        {
            ans=multi(ans,a);
        }
        k>>=1;
        a=multi(a,a);
    }
    return ans;
}
int main()
{
    int tt=1;
    while(~scanf("%lld%lld",&n,&k))
    {
        Matrix A;
        Input();
        if(n==1)
        {
            printf("Case #%d: %lld\n",tt++,k+1);
            continue;
        }
        for(int i=0;i<k;i++)
            for(int j=0;j<k;j++)
            {
                if(i==j+1)
                    A.m[i][j]=k+1-i;
                else if(i<=j)
                    A.m[i][j]=1;
                else
                    A.m[i][j]=0;
            }
       /* for(int i=0;i<k;i++)
        {
            for(int j=0;j<k;j++)
                printf("%d ",A.m[i][j]);
            cout <<endl;
        }*/
        Matrix ans=quick_mod(A,n-1);
        /*for(int i=0;i<k;i++)
        {
            for(int j=0;j<k;j++)
                printf("%lld ",ans.m[i][j]);
            cout <<endl;
        }*/
        LL t=0;
        for(int i=0;i<k;i++)
            t+=ans.m[i][0]*(k+1)%MOD;
        printf("Case #%d: %lld\n",tt++,t%MOD);
    }
    return 0;
}