http://codeforces.com/problemset/problem/514/E
大意:
推广转移矩阵,人为设定其规模是百行百列:
大意:
给出一颗树,父节点到子节点有不同的路径,求解离根节点的距离不大于x的节点数(含根结点)
分析:设距离是i的结点个数是dp[i],父节点到子节点距离是i的路径的条数是cnt[i]。
那么对于图中的转移矩阵就是:
那么对于图中的转移矩阵就是:
推广转移矩阵,人为设定其规模是百行百列:
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <set>
#include <cstring>
using namespace std;
typedef long long LL;
const int N=105,M=1e5+10,mod=1e9+7;
LL d[M];
LL cnt[N];
LL dp[N];
struct matrix{
LL m[N][N];
}A,I;
matrix multi(matrix a,matrix b){
matrix ans;
for(int i=0;i<N;i++){
for(int j=0;j<N;j++){
ans.m[i][j]=0;
for(int k=0;k<N;k++){
ans.m[i][j]=(ans.m[i][j]+a.m[i][k]*b.m[k][j]%mod)%mod;
}
}
}
return ans;
}
matrix power(matrix a,int p){
matrix ans=I;
while(p){
if(p&1) ans=multi(ans,a);
a=multi(a,a);
p>>=1;
}
return ans;
}
int main()
{
for(int i=0;i<N;i++){
I.m[i][i]=1;
}
LL n,x;
while(~scanf("%I64d%I64d",&n,&x)){
memset(cnt,0,sizeof(cnt));
memset(dp,0,sizeof(dp));
memset(A.m,0,sizeof(A.m));
//LL max_dex=0;
for(int i=0;i<n;i++) {
scanf("%I64d",&d[i]);
cnt[d[i]]++;
//max_dex=max(max_dex,d[i]);
}
dp[0]=1;
for(int i=1;i<N;i++){
for(int j=1;j<=i;j++){
dp[i]=(dp[i]+cnt[j]*dp[i-j]%mod)%mod;
}
}
LL sum=0;
if(x<=100){
for(int i=0;i<=x;i++){
sum=(sum+dp[i])%mod;
}
printf("%I64d\n",sum);
continue;
}
for(int i=0;i<=100;i++){
sum=(sum+dp[i])%mod;
}
for(int i=2;i<=100;i++){
A.m[i-2][i-1]=1;
}
for(int i=1;i<=100;i++){
A.m[99][i-1]=cnt[101-i];
A.m[100][i-1]=cnt[101-i];
}
A.m[100][100]=1;
A=power(A,x-100);
LL ans=0;
for(int i=0;i<100;i++){
ans=(ans+A.m[100][i]*dp[i+1]%mod)%mod;
}
ans=(ans+A.m[100][100]*sum%mod)%mod;
printf("%I64d\n",ans);
}
return 0;
}