2017CCPC杭州 - D题 递推计数

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6267
题目大意：


然后分别维护上一行的个数和sum[]和上一行的和s[]。递推就可以了。

#include <bits/stdc++.h>
#define LL long long
using namespace std;
const int mod = 998244353;

LL s[100005], sum[100005], T[100005], a[100005];

LL quick_pow(LL a, LL b)
{
    LL ans=1;
    while(b)
    {
        if(b&1)
            ans=ans*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return ans;
}

int main()
{
    T[1]=1;
    for(int i=2; i<=100005; i++){
        T[i]=(T[i-1]*i)%mod;
    }
    int t;scanf("%d", &t);

    while(t--){
        int n;scanf("%d", &n);
        for(int i=1; i<=n; i++){
            scanf("%lld", &a[i]);
        }
        s[1]=a[1], sum[1]=1;
        for(int i=2; i<=n; i++){
            sum[i]=(sum[i-1]+T[i-1])%mod;
            s[i]=((s[i-1]*(i-1))%mod+(sum[i]*a[i])%mod)%mod;
            sum[i]+=(sum[i-1]*(i-1))%mod;
            sum[i]%=mod;
        }
        printf("%d\n", s[n]*quick_pow(T[n], mod-2)%mod);

    }

    return 0;
}