[学习笔记] Berlekamp-Massey算法 - 学习笔记

重新实现了一个看上去就像是对的的东西。
推荐：传送门
讲的很清楚了，不多赘述。

#include<bits/stdc++.h>
#define gc getchar()
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define Rep(i,v) rep(i,0,(int)v.size()-1)
#define lint long long
#define mod 998244353
#define db double
#define pb push_back
#define mp make_pair
#define fir first
#define sec second
#define debug(x) cerr<<#x<<"="<<x
#define sp <<" "
#define ln <<endl
using namespace std;
typedef pair<int,int> pii;
typedef set<int>::iterator sit;
inline int inn()
{
	int x,ch;while((ch=gc)<'0'||ch>'9');
	x=ch^'0';while((ch=gc)>='0'&&ch<='9')
		x=(x<<1)+(x<<3)+(ch^'0');return x;
}
const int N=100000;vector<int> r[N];int d[N],a[N],cnt,fail[N];
inline int fast_pow(int x,int k,int ans=1) { for(;k;k>>=1,x=(lint)x*x%mod) (k&1)?ans=(lint)ans*x%mod:0;return ans; }
int main()
{
	int n=inn();rep(i,1,n) a[i]=inn();
	rep(i,1,n)
	{
		int &t=d[i]=a[i];Rep(j,r[cnt]) t-=(lint)a[i-j-1]*r[cnt][j]%mod,(t<0?t+=mod:0);
		if(!t) continue;fail[cnt]=i;if(!cnt) { r[++cnt].resize(i,0);continue; }
		int v=(lint)d[i]*fast_pow(d[fail[cnt-1]],mod-2)%mod,u;
		r[cnt+1].resize(i-fail[cnt-1]-1,0),r[cnt+1].pb(v);
		Rep(j,r[cnt-1]) u=(lint)v*r[cnt-1][j]%mod,r[cnt+1].pb(u?mod-u:0);
		cnt++;if(r[cnt].size()<r[cnt-1].size()) r[cnt].resize(r[cnt-1].size());
		Rep(j,r[cnt-1]) r[cnt][j]+=r[cnt-1][j],(r[cnt][j]>=mod?r[cnt][j]-=mod:0);
	}
	Rep(i,r[cnt]) cerr<<r[cnt][i]sp;cerr ln;return 0;
}