dtoj3226荒芜废墟(ruins)
荒芜废墟(ruins)
Ori来到了荒芜废墟,荒芜废墟可以看成一条数轴。Ori一开始在 $ 0 $ ,每一次可以往右跳 $ \leq k $ 的步。但有些位置设置了激光,这些位置Ori是无法到达的,一共有 $ m $ 种激光,每一种激光有一个特征值 $ t_i $ 表示这种激光会出现在 $ t_i $ 以及 $ t_i $ 的倍数的位置上。Ori想要知道从 $ 0 $ 走到 $ n $ 有多少种走法。
对于100%的数据, $ n\leq10^9 $ , $ m\leq3,k\leq50 $ ,保证ti,k可以写成 $ 2^a*3^b $ 的形式,其中 $ a,b\leq5 $
Sol
n是1e9,我们考虑矩乘来转移。
注意到 $ 2^a*3^b \leq 3125$
也就是说ti和k的lcm最多3125,记为T
前k步的转移是有限制的,不能从转i<0转,需要特殊处理
处理完我们就T步T步的转移,每T步的转移都是相同的。
最后一段也暴力转移即可。
#include<cstdio> #include<iostream> #include<cstdlib> #include<cstring> #include<algorithm> #include<cmath> #define ll long long #define mod 1000000007 using namespace std; int n,m,k,t[4],T; ll f[55],F[8005],a[8500][55]; struct node{ ll v[52][52]; }p,ans; node ch(node a,node b){ node c; for(int i=0;i<k;i++) for(int j=0;j<k;j++){ c.v[i][j]=0; for(int x=0;x<k;x++){ c.v[i][j]=(c.v[i][j]+a.v[i][x]*b.v[x][j]%mod)%mod; } } return c; } int main(){ cin>>n>>m>>k;T=k; for(int i=1;i<=m;i++){ scanf("%d",&t[i]); T=T*t[i]/__gcd(T,t[i]); } f[0]=1;a[0][0]=1; for(int i=1;i<=k-1;i++){ a[i][i]=1; for(int j=i-1;j>=0;j--){ f[i]=(f[i]+f[j])%mod; } for(int x=1;x<=m;x++)if(i%t[x]==0)f[i]=0; } for(int i=k;i<T+k;i++){ for(int j=i-1;j>=i-k;j--){ int fl=0;for(int x=1;x<=m;x++)if(j%t[x]==0)fl=1; if(fl&&j)continue; for(int x=0;x<k;x++){ a[i][x]=(a[i][x]+a[j][x])%mod; } } for(int j=1;j<=m;j++){ if(i%t[j]==0)for(int x=0;x<k;x++)a[i][x]=0; } } for(int i=T;i<T+k;i++){ for(int j=0;j<k;j++)p.v[j][i-T]=a[i][j]; ans.v[i-T][i-T]=1; } int num=n/T;n%=T; while(num){ if(num&1)ans=ch(ans,p); p=ch(p,p);num>>=1; } for(int i=0;i<k;i++){ for(int j=0;j<k;j++){ F[i]=(F[i]+ans.v[j][i]*f[j]%mod)%mod; } } for(int i=k;i<=n;i++){ for(int j=i-1;j>=i-k;j--){ F[i]=(F[i]+F[j])%mod; } for(int x=1;x<=m;x++)if(i%t[x]==0)F[i]=0; } cout<<F[n]<<endl; return 0; }