2021牛客多校1——I：Increasing Subsequence（动态规划）

题面

题意：

• 给出排列P，两个人轮流取数，每次取的数需要在之前该人取数的右边，且比当前取出来所有的数都要大。所有当前可选的数都将等概率随机的被当前决策人选中。问两个人期望取数的轮数。
• 规则1：每个人取的数字都应该比当前已经被取出来的数字大。
• 规则2：每个人取的数字都只能取他上一个取的数字右边的数字。

思路:

• 通过两个规则，我们可以依次用dp[i][j]来代表前一个取了i，后一个人取了j，因为它是一个长度为n的排列，因此我们可以利用逆序的下标来满足规则1，我们先用i遍历一遍下标n到1，再逆序遍历数组a，来满足规则2， 我们只有在a[j]小于i的时候才满足规则1，因此判断当前情况下的概率，如果a[j]大于i,则将dp[i][a[j]]的概率加起来，同时记录当前有多少个大于i的数字，之后再小于i的a[j]的时候取大于i的a[j]就满足题意，因此dp[a[j]][i]往下取的概率则为总概率除以比当前位下标大并值大的个数。通过状态转移即可得到答案。

代码:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll MAXN = 5e3 + 10;
const ll mod = 998244353;
ll a[MAXN];
ll dp[MAXN][MAXN];
ll qpow(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() {
    ll n;
    scanf("%lld", &n);
    for (ll i = 1; i <= n; i++) {
        scanf("%lld", &a[i]);
    }
    for (ll i = n; i >= 1; i--) {
        ll cnt = 0;
        ll sum = 0;
        for (ll j = n; j >= 0; j--) {
            if (a[j] < i) {
                dp[a[j]][i] = 1;
                if (cnt) {
                    dp[a[j]][i] += sum * qpow(cnt, mod - 2);
                }
            }
            if (a[j] > i) {
                sum = (sum + dp[i][a[j]]) % mod;
                cnt++;
            }
        }
    }
    ll ans = 0;
    for (ll i = 1; i <= n; i++) {
        ans = (ans + dp[0][i]) % mod;
    }
    printf("%lld\n", ans * qpow(n, mod - 2) % mod);
    return 0;
}