1. 程式人生 > >[學習筆記] Berlekamp-Massey演算法 - 學習筆記

[學習筆記] 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; }