在很多题目总我们列出了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;
}