[學習筆記] Berlekamp-Massey演算法 - 學習筆記
阿新 • • 發佈:2018-11-01
重新實現了一個看上去就像是對的的東西。
推薦:傳送門
講的很清楚了,不多贅述。
#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;
}