1. 程式人生 > >hdu6267 Master of Random(期望)

hdu6267 Master of Random(期望)

比賽的時候是隊友推的。但是賽後想了想這種題找到方向了,就不該推那麼長時間。

思路:

一看,所有情況都列舉一遍顯然是不合理的 。那麼我就去轉化思維,想每一個點對總答案的貢獻。(很多題都是這個想法。)

推導過程:

還要乘以權值。

程式碼:

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const LL mod=998244353;
LL quick(LL a,LL b){
    LL ans=1;
    while(b){
        if(b&1) ans=(ans*a)%mod;
        b>>=1;
        a=(a*a)%mod;
    }
    return ans;
}
LL inv(LL x){
    return quick(x,mod-2);
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--){
        int n;
        scanf("%d",&n);
        LL tp=1;
        for(int i=1;i<n;i++){
            tp=(tp*i)%mod;///計算(n-1)!
        }
        LL a,tmp=tp;
        scanf("%lld",&a);
        LL ans=tp*a%mod;
        for(int i=1;i<n;i++){
            scanf("%lld",&a);
            tmp=tmp+tp*inv(i)%mod;
            ans=(ans+tmp*a)%mod;
        }
        tp=tp*n%mod;
        printf("%lld\n",ans*inv(tp)%mod);
    }
    return 0;
}