1. 程式人生 > >myh的超級多項式

myh的超級多項式

題目大意: 給你n,k,{vk},{fk}n,k,\{v_k\},\{f_k\},並且已知fn=i=1kaivinf_n=\sum_{i=1}^k a_iv_i^{n},求fn,nk1000,k105,O(k(nk))f_n,n-k\le1000,k\le10^5,O(k(n-k))可過。 題解:考慮構造fnf_n關於fnkfn1f_{n-k}\ldots f_{n-1}的遞推式,也就是要求係數序列{ck}\{c_k\},使得:fn=i=1kc

ifnif_n=\sum_{i=1}^kc_if_{n-i}成立。 令c0=1c_0=-1,等價於: i=0kcifni=0i=0kcij=1kajvjni=0\sum_{i=0}^kc_if_{n-i}=0\\ \sum_{i=0}^kc_i\sum_{j=1}^ka_jv_j^{n-i}=0 其一個充分條件是: j[1,k],i=0kciajvjni=0\forall j\in[1,k],\sum_{i=0}^kc_ia_jv_j^{n-i}=0 亦即: j[1,k],i=0kcivjki=0\forall j\in[1,k],\sum_{i=0}^kc_iv_j^{k-i}=0F(x)=i=0kckixiF(x)=\sum_{i=0}^kc_{k-i}x^i,則上式等價於{vk}\{v_k\}F(x)F(x)的零點,再由[xk]F(x)=c0=1[x^k]F(x)=c_0=-1可知:F(x)=i=1k(xvi)F(x)=-\prod_{i=1}^k(x-v_i) 使用分治FFT求出F(x)F(x)進而得到{ck}\{c_k\}進而求出fnf_n即可。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
#define p 1004535809
#define N 100010
#define lint long long
#define gc getchar()
#define clr(a,n) memset(a,0,sizeof(int)*(n))
using namespace std;
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;
}
int a[N<<2],b[N<<2],r[N<<2],c[N],x[N],v[N];vector<int> f[N];
inline int fast_pow(int x,int k,int ans=1) { for(;k;k>>=1,x=(lint)x*x%p) (k&1)?ans=(lint)ans*x%p:0;return ans; }
inline int NTT(int *a,int n,int s)
{
    for(int i=1;i<n;i++) if(i<r[i]) swap(a[i],a[r[i]]);
    for(int i=2;i<=n;i<<=1)
    {
        int wn=fast_pow(3,s>0?(p-1)/i:p-1-(p-1)/i);
        for(int j=0,t=i>>1;j<n;j+=i) for(int k=0,w=1,x,y;k<t;k++,w=(lint)w*wn%p)
            x=a[j+k],y=(lint)w*a[j+k+t]%p,((a[j+k]=x+y)>=p?a[j+k]-=p:0),((a[j+k+t]=x-y)<0?a[j+k+t]+=p:0);
    }
    if(s<0) { int ninv=fast_pow(n,p-2);for(int i=0;i<n;i++) a[i]=(lint)a[i]*ninv%p; }
    return 0;
}
inline int tms(const vector<int> &A,const vector<int> &B,vector<int> &C)
{
    int m1=(int)A.size(),m2=(int)B.size();
    int n=1,L=0;for(;n<m1+m2-1;n<<=1,L++);
    for(int i=1;i<n;i++) r[i]=(r[i>>1]>>1)|((i&1)<<(L-1));
    clr(a,n);for(int i=0;i<m1;i++) a[i]=A[i];NTT(a,n,1);
    clr(b,n);for(int i=0;i<m2;i++) b[i]=B[i];NTT(b,n,1);
    for(int i=0;i<n;i++) a[i]=(lint)a[i]*b[i]%p;NTT(a,n,-1);
    C.resize(m1+m2-1);for(int i=0;i<m1+m2-1;i++) C[i]=a[i];
    return 0;
}
int solve(int l,int r)
{
    if(l==r) return f[l].resize(2),f[l][0]=(v[l]?p-v[l]:0),f[l][1]=1;
    int mid=(l+r)>>1;solve(l,mid),solve(mid+1,r),tms(f[l],f[mid+1],f[l]);
    return 0;
}
int main()
{
    int n=inn(),k=inn();
    for(int i=1;i<=k;i++) v[i]=inn();
    for(int i=1;i<=k;i++) x[i]=inn(