牛客挑战赛48 B-心跳调试 概率线性DP

传送门： https://ac.nowcoder.com/acm/contest/11161/B

题意

思路

$假设栈顶为i，那么下一个入栈的必须是i或者i+1，所以和栈内的状态无关。$

$设当前为i，下一个放入i+1而不是更大的数字的概率为p，则$
$$p=f_i\ast p+f_{i+1}$$
$则：$
$$p=\frac{w_{i+1}}{{\sum }_{k=i+1}^n{w}_k}$$

$则:$
$$ans=\prod _{i=2}^n{p}_i$$

$复杂度O(n)dp。$

Code

#include "bits/stdc++.h"
using namespace std;

typedef long long ll;
const ll mod = 1e9 + 7;

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 % mod;
}

void solve() {
    int n; cin >> n;
    vector<ll> w(n + 100), sum(n + 100);
    for(int i = 1;i <= n; i++) cin >> w[i];
    for(int i = n;i >= 1; i--) sum[i] = sum[i + 1] + w[i];
    ll ans = 1;
    for(int i = 2;i <= n; i++) {
        ans = ans * w[i] % mod * quick_pow(sum[i] % mod, mod - 2) % mod;
    }
    cout << ans << endl;
}

signed main() {
    solve();
}