Educational Codeforces Round 83 (Rated for Div. 2)D. Count the Arrays

D. Count the Arrays

标签：组合数学

题意：给定n和m，计算出满足下列条件的序列的个数（答案对998244353取模）：

• 每个序列包括n个元素
• 每个元素都是1到m的数
• 每个序列中必须存在且仅存在一对相等的数
• 对于一个序列，应当存在一个下标i，满足i的左侧严格递增，i的右侧严格递减

题解

既然需要满足的条件都已经列出来了，那我们就一条条看

首先序列包括1到m的n个元素并且必须存在唯一一对相等的数，那么就需要在m个数中选出n - 1个数，共有$C_m^{n-1}$种选法

然后是选出在两侧的数，结合最后一个条件我们知道最大的数一定不能放在两侧，那么实际上就是n - 2个数种选一个数，共有n - 2种选法

最后一个条件考虑最大的数所在位置，按照这个数前面有几个数来计算（选定了前面的数，后面的数也就确定了），由于要满足单调性的条件，所有我们只是选出这些数，而不需要管他们的排列，去除两侧的数和最大的数，我们要在n - 3个数中进行选择，一共有$C_{n-3}^0$ + $C_{n-3}^1$ + … + $C_{n-3}^{n-3}$ = $2^{n-3}$种

最后的答案就是将以上结果相乘即可，注意n = 2的情况特判一下

代码实现

#include <algorithm>
#include <iostream>
#include <stdio.h>
#include <iomanip>
#include <cstring>
#include <string>
#include <vector>
#include <queue>
#include <set>
#include <map>
#define ll long long
using namespace std;
const int mod = 998244353;
ll n, m;
ll qpow(ll x, ll y){
    x %= mod;
    ll ans = 1;
    while(y){
        if(y & 1) ans = ans * x % mod;
        x = x * x % mod;
        y >>= 1;
    }
    return ans;
}
ll inv(ll x){
    return qpow(x, mod - 2) % mod;
}
ll cal(ll m, ll n){
    if(m > n) return 0;
    ll up = 1, down = 1;
    for(int i = 0; i < m; i++){
        up = up * (n - i) % mod;
        down = down * (i + 1) % mod;
    }
    return up * inv(down) % mod;
}
int main(){
    cin >> n >> m;
    if(n == 2) cout << 0;
    else{
        ll ans = cal(n - 1, m) * (n - 2) % mod * qpow(2, n - 3) % mod;
        cout << ans; 
    }
    return 0;
}