题意
有一棵n叉树,深度是无限的,每个结点有n个儿子。从左到右编号为1到n号儿子,第i号儿子离该结点的距离是di。现在要统计一下距离根结点不超过x的结点有多少个。
数字比较大对 109 + 7 取余后输出。
样例解释:
图中黄色的结点是距离根不超3的。
1≤n≤10^5,0≤x≤10^9,1<=di<=100
分析
注意到di很小,考虑设f[i]表示离根节点距离为i的节点数量。用矩阵快速幂优化转移即可。
代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const int MOD=1000000007;
int n,m,d[100005],mx;
struct Matrix
{
int a[105][105];
void clear(int n)
{
for (int i=1;i<=n;i++)
for (int j=1;j<=n;j++)
a[i][j]=0;
}
void unit(int n)
{
clear(n);
for (int i=1;i<=n;i++) a[i][i]=1;
}
}a;
void mul(Matrix &c,Matrix a,Matrix b)
{
c.clear(mx);
for (int i=1;i<=mx;i++)
for (int k=1;k<=mx;k++)
for (int j=1;j<=mx;j++)
(c.a[i][j]+=(LL)a.a[i][k]*b.a[k][j]%MOD),
c.a[i][j]-=c.a[i][j]>=MOD?MOD:0;
}
Matrix ksm(Matrix x,int y)
{
Matrix ans;ans.unit(mx);
while (y)
{
if (y&1) mul(ans,ans,x);
mul(x,x,x);y>>=1;
}
return ans;
}
int main()
{
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++) scanf("%d",&d[i]),mx=max(mx,d[i]);
for (int i=1;i<=n;i++) a.a[mx-d[i]+1][mx]++,a.a[mx-d[i]+1][mx+1]++;
for (int i=2;i<=mx;i++) a.a[i][i-1]++;
mx++;a.a[mx][mx]++;
a=ksm(a,m);
printf("%d",(a.a[mx][mx]+a.a[mx-1][mx])%MOD);
return 0;
}