bzoj 4929: 第三题

Description

给定n,b,c,d,e以及A0,A1,···,An−1，定义

x_k=b×c^^4k+d×c^^2k+e

f(x)=Sigma(A_ix^ⁱ),0<=i<=n-1

请你求出f(x₀),f(x₁),···,f(x_n−1)对10^6+3取模的值。

Input

第一行包括五个整数n,b,c,d,e。

接下来一行包括n个整数，代表a₀,a₁,···,a_n−1。

N<=60000,保证给出的数字都为整数且均在 [0, 10^6]

Output

N行，第i行代表f(x_i−1)

多项式多点求值可过，但似乎不是正解

#include<bits/stdc++.h>
typedef std::vector<int> pol;
typedef long long i64;
typedef unsigned long long u64;
const int P=;
int max(int a,int b){return a>b?a:b;}
template<int p>
i64 pw(int a,int n){
    if(n<)n+=p-;
    i64 v=;
    for(;n;n>>=,a=i64(a)*a%p)if(n&)v=v*a%p;
    return v;
}
int N,K,r[<<|];
typedef double ld;
struct cplx{
    ld a,b;
    cplx(ld _x=,ld _y=):a(_x),b(_y){}
    cplx operator+(cplx w){return cplx(a+w.a,b+w.b);}
    cplx operator-(cplx w){return cplx(a-w.a,b-w.b);}
    cplx operator*(cplx w){return cplx(a*w.a-b*w.b,a*w.b+b*w.a);}
}A[<<|],B[<<|],*Es[][],C[<<|],D[<<|];
void pre(int n){
    for(N=,K=;N<n;N<<=,++K);
    for(int i=;i<N;++i)r[i]=r[i>>]>>|(i&)<<K;
    memset(A,,sizeof(cplx)*N);
    memset(B,,sizeof(cplx)*N);
    memset(C,,sizeof(cplx)*N);
    memset(D,,sizeof(cplx)*N);
}
const ld pi=acos(-);
void dft(cplx*a,int t){
    for(int i=;i<N;++i)if(i<r[i])std::swap(a[i],a[r[i]]);
    for(int i=,z=;i<N;i<<=,++z){
        if(!Es[t==][z]){
            cplx*E=Es[t==][z]=new cplx[i];
            for(int j=;j<i;++j)E[j]=cplx(cos(j*pi/i),t*sin(j*pi/i));
        }
        cplx*E=Es[t==][z];
        for(int j=;j<N;j+=i<<){
            cplx*b=a+j,*c=b+i;
            for(int k=;k<i;++k){
                cplx x=b[k],y=c[k]*E[k];
                b[k]=x+y;
                c[k]=x-y;
            }
        }
    }
    if(t==-)for(int i=;i<N;++i)a[i].a/=N;
}
void mul(cplx*a,cplx*b){
    for(int i=;i<N;++i)a[i]=a[i]*b[i];
}
void mov(const pol&x,cplx*a){
    for(int i=;i<x.size();++i)a[i].a=x[i];
}
void mov(cplx*a,const pol&_x){
    pol&x=const_cast<pol&>(_x);
    for(int i=;i<x.size();++i)x[i]=i64(a[i].a+0.49)%P;
}
void mov(const pol&x,int*a){
    memcpy(a,x.data(),x.size()*sizeof(int));
}
void mov(int*a,const pol&_x){
    pol&x=const_cast<pol&>(_x);
    memcpy(x.data(),a,x.size()*sizeof(int));
}
void chk(pol&w){
    int p=w.size();
    while(p&&!w[p-])--p;
    w.resize(p);
}
void rev(pol&w){
    std::reverse(w.data(),w.data()+w.size());
}
u64 _c[];
pol operator*(const pol&a,const pol&b){
    pol c(a.size()+b.size()-);
    if(c.size()<=){
        const int*as=a.data(),ap=a.size(),*bs=b.data(),bp=b.size();
        int*cs=c.data();
        for(int i=;i<c.size();++i)_c[i]=;
        for(int i=;i<ap;++i)for(int j=;j<bp;++j)_c[i+j]+=u64(as[i])*bs[j];
        for(int i=;i<c.size();++i)cs[i]=_c[i]%P;
    }else if(c.size()<=){
        pre(c.size());
        mov(a,A),mov(b,B);
        dft(A,),dft(B,);
        mul(A,B);
        dft(A,-);
        mov(A,c);
    }else{
        pre(c.size());
        for(int i=;i<a.size();++i){
            A[i]=a[i]>>;
            B[i]=a[i]&;
        }
        for(int i=;i<b.size();++i){
            C[i]=b[i]>>;
            D[i]=b[i]&;
        }
        dft(A,);dft(B,);dft(C,);dft(D,);
        for(int i=;i<N;++i){
            cplx v1=A[i]*C[i];
            cplx v2=A[i]*D[i]+B[i]*C[i];
            cplx v3=B[i]*D[i];
            A[i]=v1;
            B[i]=v2;
            C[i]=v3;
        }
        dft(A,-);dft(B,-);dft(C,-);
        for(int i=;i<c.size();++i){
            c[i]=((i64(A[i].a+0.49)<<)+(i64(B[i].a+0.49)<<)+i64(C[i].a+0.49))%P;
        }
    }
    chk(c);
    return c;
}
pol operator*(const pol&w,int x){
    pol a(w);
    for(int i=;i<a.size();++i)a[i]=i64(a[i])*x%P;
    return a;
}
pol operator-(const pol&a,const pol&b){
    pol c(max(a.size(),b.size()));
    mov(a,c.data());
    for(int i=;i<b.size();++i)c[i]=(c[i]-b[i]+P)%P;
    chk(c);
    return c;
}
pol inv(pol a){
    if(a.size()==){
        a.resize();
        a[]=pw<P>(a[],P-);
        return a;
    }
    pol b=inv(pol(a.data(),a.data()+(a.size()-)/+));
    pol c=b*b*a;
    c=b*-c;
    c.resize(a.size());
    return c;
}
pol operator/(pol a,pol b){
    chk(a),chk(b);
    rev(a),rev(b);
    int sz=a.size()-b.size()+;
    b.resize(sz);
    pol c=a*inv(b);
    c.resize(sz);
    rev(c);
    chk(c);
    return c;
}
pol operator%(const pol&a,const pol&b){
    if(a.size()<b.size())return a;
    return a-a/b*b;
}
int cal(const pol&w,i64 x){
    int y=;
    for(int p=w.size()-;p>=;--p)y=(y*x+w[p])%P;
    return y;
}
int n,b,c,d,e,a[],as[];
struct Q{int x,id;}qs[];
bool operator<(Q a,Q b){return a.x<b.x;}
pol tr[];
void calc0(int w,int L,int R){
    if(L==R){
        tr[w]=pol();
        tr[w][]=;
        tr[w][]=(P-qs[L].x)%P;
        return;
    }
    int M=(L+R)/;
    calc0(w<<,L,M);
    calc0(w<<^,M+,R);
    tr[w]=tr[w<<]*tr[w<<^];
}
void calc1(const pol&a,int w,int L,int R){
    if(R-L<=){
        for(int i=L;i<=R;++i)as[qs[i].id]=cal(a,qs[i].x);
        return;
    }
    int M=(L+R)/;
    calc1(a%tr[w<<],w<<,L,M);
    calc1(a%tr[w<<^],w<<^,M+,R);
}
int main(){
    scanf("%d%d%d%d%d",&n,&b,&c,&d,&e);
    for(int i=;i<n;++i)scanf("%d",a+i);
    for(int i=;i<n;++i){
        qs[i].x=(b*pw<P>(c,*i)+d*pw<P>(c,*i)+e)%P;
        qs[i].id=i;
    }
    std::sort(qs,qs+n);
    tr[]=pol(n);
    mov(a,tr[]);
    calc0(,,n-);
    calc1(tr[],,,n-);
    for(int i=;i<n;++i)printf("%d\n",(as[i]+P)%P);
    return ;
}