关于矩阵优化的DP总结

在很多题目总我们列出了dp的转移状态方程，但是某一维的之很大，用滚动数组会超时，这是我们就将借矩阵来进行优化，利用矩阵优化的关键就是构造A矩阵，自己期初不能很轻易的列出矩阵，通过一些列的研究，自己总结了一些方法。

1.当后一项至于前n项有关时，我们就建立一个n*n的矩阵

例如斐波拉契数列利用矩阵来求第n相

An-1      0 1         An-1

An         1 1         An-2

构造出来的矩阵为0 1

                             1 1

然后只需求矩阵的n次方即可

hdu2294

本题列出的状态转移方程dp[i][j]=dp[i-1][j]*j+dp[i-1][j-1]*(k-j+1)。

由于此题中n很大，所以不能直接转移，因此需要我们利用矩阵来进行优化，现在最足要的就是进行矩阵的构造

dp[i-1][o],dp[i-1][1]............dp[i-1][k].

构造举证利用上数斐波拉契数列的方法构造一个（k+1）*（k+1）的矩阵即：

0 k 0 0........0

0 1 k-1 ........

0 0  2  .........

.

.

.

0 0 0 0 0.....k 

代码：

#include<cstdio>
#include<set>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 35
#define mod 1234567891
typedef long long LL;

int k;
struct Matrix
{
    LL a[N][N];
    int nn;
    void init(int n)
    {
        nn=n;
        for(int i=0; i<n; i++)
        {
            for(int j=0; j<n; j++)
            {
                a[i][j]=0;
            }
        }
    }
    Matrix operator + (const Matrix &b) const
    {
        Matrix tem;
        tem.nn=b.nn;
        for(int i=0; i<b.nn; i++)
        {
            for(int j=0; j<b.nn; j++)
            {
                tem.a[i][j]=(a[i][j]%mod+b.a[i][j]%mod)%mod;
            }
        }
        return tem;
    }
    Matrix operator *(const Matrix &b) const
    {
        Matrix tem;
        tem.nn=b.nn;
        for(int i=0; i<b.nn; i++)
        {
            for(int j=0; j<b.nn; j++)
            {
                tem.a[i][j]=0;
                for(int k=0; k<b.nn; k++)
                {
                    tem.a[i][j]=(tem.a[i][j]%mod+(a[i][k]%mod*b.a[k][j]%mod)%mod)%mod;
                }
            }
        }
        return tem;
    }
    Matrix & operator = (const Matrix & b)
    {
        this->nn=b.nn;
        for(int i=0; i<b.nn; i++)
        {
            for(int j=0; j<b.nn; j++)
            {
                this->a[i][j]=b.a[i][j];
            }
        }
        return *this;
    }
} A,E;
Matrix MatrixPow(Matrix ret,int n)
{
    Matrix ans;
    ans.init(k + 1);
    for (int i = 0; i < ans.nn; ++i)
    {
        for (int j = 0; j < ans.nn; ++j)
        {
            ans.a[i][j] = (i == j);
        }
    }
    while(n)
    {
        if (n & 1)
        {
            ans = ans * ret;
        }
        n >>= 1;
        ret = ret * ret;
    }
    return ans;
}
Matrix slove(int cnt)
{
    if (cnt == 1)
    {
        return A;
    }
    Matrix tmp = slove(cnt >> 1);
    Matrix c = MatrixPow(A, cnt >> 1);
    c = c + E;
    tmp = c * tmp;
    if (cnt & 1)
    {
        Matrix d = MatrixPow(A, cnt);
        tmp = tmp + d;
    }
    return tmp;
}
int main()
{
    int T,n;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&k);
        A.init(k+1);
        E.init(k+1);
        for(int i=0; i<E.nn; i++)
        {
            for(int j=0; j<E.nn; j++)
            {
                E.a[i][j]=(i==j);
            }
        }
        A.a[0][0]=0;
        A.a[0][1]=k;
        for(int i=1; i<A.nn; i++)
        {
            A.a[i][i]=i;
            if(i+1<A.nn)
                A.a[i][i+1]=k-i;
        }
        Matrix ans;
        ans.init(k+1);
        ans=slove(n);
        for(int i=0;i<=k;i++)
        {
            printf("%lld\n",ans.a[i][k]);
        }
    }
    return 0;
}