codeforces852F Product transformation 排列组合

题目链接：戳这里

题目大意：有一个长度为n的数组A，开始时，A内的元素全为a。

现在进行m次操作，每次操作将A[i]变为A[i]*A[i+1]，最后一个元素不变。

现给出n,m,a,Q，输出m次操作后的A数组，每个元素对Q取模。

题解：因为每个数刚开始都是一样的，那么我们可以把乘法考虑为指数的相加。

用S代表指数，那么Si=Si+S(i+1)。

设S[i][j]表示第i个数经过j次操作后的次数，那么显然S[i][j]=S[i-1][j]+S[i-1][j+1]。

好像很眼熟？很像杨辉三角的递推式吧。

那么我们将序列翻转，那么显然S[i][j]=S[i-1][j]+S[i-1][j-1]，好了这就是组合数。

那么我们就直接统计答案即可。

代码：

#include<bits/stdc++.h>
#define maxn 1000500
using namespace std;
typedef long long LL;
LL read()
{
    char c;LL sum=0,f=1;c=getchar();
    while(c<'0' || c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0' && c<='9'){sum=sum*10+c-'0';c=getchar();}
    return sum*f;
}
LL n,m,a;
LL ans[maxn],fac[maxn],inv[maxn];
LL mod,p;
LL ksm(LL a,LL b,LL mod)
{
    LL ret=1;
    while(b)
    {
        if(b&1) ret=ret*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return ret;
}
void init()
{
    fac[0]=1;
    for(int i=1;i<=m;i++) fac[i]=fac[i-1]*i%p;
    inv[m]=ksm(fac[m],p-2,p);
    for(int i=m-1;i>=0;i--)
    inv[i]=inv[i+1]*(i+1)%p;
}
LL C(LL n,LL m)
{
    return n<m?0:fac[n]*inv[m]%p*inv[n-m]%p;
}
int main()
{
    n=read();m=read();a=read();mod=read();
    LL now=1;
    for(LL i=1;;i++)
    {
        now=now*a%mod;
        if(now==1){p=i;break;}
    }
    init();
    for(int i=1;i<=n;i++)
    ans[i]=C(m,i-1);
    for(int i=1;i<=n;i++)
    ans[i]=(ans[i]+ans[i-1])%p;
    for(int i=1;i<=n;i++)
    ans[i]=ksm(a,ans[i],mod);
    for(int i=n;i>=1;i--)
    printf("%lld ",ans[i]);
    return 0;
}