codeforces 514E E. Darth Vader and Tree（矩阵应用）

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;
}