【BZOJ4161】Shlw loves matrixI (常系数齐次线性递推)
线性 递推 系数 loves
2023-09-11 14:14:40 时间
【BZOJ4161】Shlw loves matrixI (常系数齐次线性递推)
题面
题解
\(k\)很小,可以直接暴力多项式乘法和取模。
然后就是常系数齐次线性递推那套理论了,戳这里
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define MOD 1000000007
#define MAX 5000
void add(int &x,int y){x+=y;if(x>=MOD)x-=MOD;}
inline int read()
{
int x=0;bool t=false;char ch=getchar();
while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
if(ch=='-')t=true,ch=getchar();
while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
return t?-x:x;
}
int fpow(int a,int b)
{
int s=1;if(a==1)return 1;
while(b){if(b&1)s=1ll*s*a%MOD;a=1ll*a*a%MOD;b>>=1;}
return s;
}
int n,k;
int tmp[MAX];
int a[MAX],h[MAX<<1];
void Multi(int *a,int *b,int n,int m,int *c)
{
for(int i=0;i<=n+m;++i)tmp[i]=0;
for(int i=0;i<=n;++i)
for(int j=0;j<=m;++j)
add(tmp[i+j],1ll*a[i]*b[j]%MOD);
for(int i=0;i<=n+m;++i)c[i]=tmp[i];
}
void Mod(int *a,int *b,int n,int m)
{
for(int i=n;i>=m;--i)
if(a[i])
{
int t=1ll*a[i]*fpow(b[m],MOD-2)%MOD;
for(int j=i;j>=i-m;--j)
add(a[j],MOD-1ll*t*b[m-i+j]%MOD);
}
}
int p[MAX],ans[MAX];
void Solve(int b,int *mod,int K,int *ans)
{
int s[MAX];memset(s,0,sizeof(s));s[1]=ans[0]=1;
while(b)
{
if(b&1)Multi(ans,s,K-1,K-1,ans),Mod(ans,p,K+K-2,K);
Multi(s,s,K-1,K-1,s);Mod(s,p,K+K-2,K);
b>>=1;
}
}
int main()
{
n=read();k=read();
for(int i=1;i<=k;++i)a[i]=(read()%MOD+MOD)%MOD;
for(int i=0;i<k;++i)h[i]=(read()%MOD+MOD)%MOD;
p[k]=1;for(int i=1;i<=k;++i)p[k-i]=(MOD-a[i])%MOD;
Solve(n-k,p,k,ans);int Ans=0;
for(int i=k;i<k+k;++i)
for(int j=1;j<=k;++j)
add(h[i],1ll*a[j]*h[i-j]%MOD);
for(int i=0;i<k;++i)add(Ans,1ll*h[i+k]*ans[i]%MOD);
printf("%d\n",Ans);
return 0;
}
相关文章
- 数据结构和算法-数据结构-线性结构-顺序表 链表和哈希表
- 数字三角形(线性DP)
- FE之DR之线性降维:PCA/白化、LDA算法的数学知识(协方差矩阵)、相关论文、算法骤、代码实现、案例应用等相关配图之详细攻略
- ML之LiR&LassoR:利用boston房价数据集(PCA处理)采用线性回归和Lasso套索回归算法实现房价预测模型评估
- FE之DR之线性降维:PCA/白化、LDA算法的数学知识(协方差矩阵)、相关论文、算法骤、代码实现、案例应用等相关配图之详细攻略
- ML之LiR&2PolyR:使用线性回归LiR、二次多项式回归2PolyR模型在披萨数据集上拟合(train)、价格回归预测(test)
- DL之RNN:基于TF利用RNN实现简单的序列数据类型(DIY序列数据集)的二分类(线性序列&随机序列)
- 基于smo算法的数据分类matlab仿真,对比线性分类和非线性分类
- 完美解决循环线性操作-倒计时-无报错
- 【数字信号处理】线性时不变系统 LTI “ 输入 “ 与 “ 输出 “ 之间的关系 ( 线性卷积起点定理 | 左边序列概念 | 推理 )
- 003-数据结构-线性结构-数组
- pyspark 线性向量机
- tensorflow 线性回归解决 iris 2分类
- 1027. 方格取数(线性DP)
- 898. 数字三角形(线性DP)