NKOI 3124 珍珠吊坠

来源：互联网发布：导航端口和波特率编辑：程序博客网时间：2024/04/28 16:44

珍珠吊坠

Time Limit:10000MS Memory Limit:65536K
Total Submit:25 Accepted:17
Case Time Limit:1000MS

Description

有k种珍珠，每种珍珠有N颗，问能组合出多少种含有k种珍珠且长度<=N的吊坠?(注意：吊坠为链状，非环状)

Input

一行，两个整数N和K

Output

一行，一个整数，表示所求方案总数,结果可能很大，mod 1234567891再输出

Sample Input

样例输入1：2 1样例输入2：3 2

Sample Output

样例输出1：2样例输出2：8

Hint

1 ≤ N ≤ 1,000,000,000
1 ≤ K ≤ 30
样例2说明：
设两种珍珠编号为1和2
1 2
2 1
1 1 2
1 2 1
2 1 1
2 2 1
2 1 2
1 2 2

Source

HDU2294

状态：
           dp[i][j]表示用j种珍珠构成长度恰好为i的吊坠，总的方案数
方程：
           dp[i][j] = (k-(j-1))*dp[i-1][j-1] + j*dp[i-1][j];
结果：
           ans[i]表示长度在i以内且满足条件的吊坠的总方案数
           ans[i]=dp[1][k]+dp[2][k]+...+dp[i][k]
           显然，问题的最终答案为ans[n]

但是考虑到n的范围，我们必须用矩阵乘法优化

根据前面的分析显然有：ans[i]=ans[i-1]+dp[i][k]
参考前面所讲的数列构造方法：
我们构造一个1*(k+1)的矩阵F{ dp[i][1], dp[i][2], dp[i][3], ... ,dp[i][k], ans[i-1] }
我们希望再构造一个(k+1)*(k+1)的矩阵A,使得:F * A={ dp[i+1][1], dp[i+1][2], dp[i+1][3], ... , dp[i+1][k], ans[i] }

矩阵A的求法很好求，这里就不过多解释了

最终答案就是F*A^n；其中矩阵F的初始值：
{dp[1][1], dp[1][2], dp[1][3], ... ,dp[1][k], ans[0]}
={ k , 0 , 0 , ... , 0 , 0 }

#include<iostream>#include<cstdio>#include<cstring>#define LL long longusing namespace std;const int mod=1234567891;typedef LL arr[35][35];arr z,A,ans;LL k,n;void cheng(arr y,arr x){memset(z,0,sizeof(z));LL i,j,q;    for(i=1;i<=k+1;i++)       for(j=1;j<=k+1;j++)          for(q=1;q<=k+1;q++)             z[i][j]=(z[i][j]+y[i][q]*x[q][j])%mod;    memcpy(y,z,sizeof(z));}void mb(LL b){LL i;for(i=1;i<=k+1;i++)ans[i][i]=1;    while(b){    if(b&1)cheng(ans,A);        b>>=1;        cheng(A,A);    }}int main(){scanf("%I64d%I64d",&n,&k);LL i,j;A[k+1][k+1]=A[k][k+1]=1,A[k][k]=k;for(i=1;i<k;i++)    A[i][i]=i,A[i][i+1]=k-i;mb(n);cout<<(k*ans[1][k+1])%mod;}

0 0