牛客多校Round 9

Solved:1

rank:112

E. Music Game

题解说有个非简化的原题 bzoj4318

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod = 1e9 + ;
ll n, m;
ll p[];
ll comb[][];

ll pow_mod(ll x, ll y)
{
    ll res = ;
    while(y)
    {
        if(y & ) res = res * x % mod;
        x = x * x % mod;
        y >>= ;
    }
    return res;
}

int main()
{
    scanf("%lld%lld", &n, &m);
    ll inv = pow_mod(, mod - );
    for(int i = ; i <= n; i++) scanf("%lld", &p[i]);
    for(int i = ; i <= n; i++)
    {
        comb[i][i] = p[i] * inv % mod;
        for(int j = i + ; j <= n; j++) comb[i][j] = comb[i][j - ] * p[j] % mod * inv % mod;
    }

    ll ans = ;
    for(int i = ; i <= n; i++)
    {
        for(int j = i; j <= n; j++)
        {
            ll tmp = pow_mod(j - i + , m);
            (tmp *= comb[i][j]) %= mod;
            if(i != ) tmp = tmp * ( - p[i - ]) % mod * inv % mod, tmp %= mod;
            if(j != n) tmp = tmp * ( - p[j + ]) % mod * inv % mod, tmp %= mod;
            (ans += tmp) %= mod;
        }
    }
    printf("%lld\n", ans);
    return ;
}