D. Count the Arrays

一道排列组合题.
思路:
1.从m个数中选出n-1个(因为有2个数相同)
2.这n - 1个数中选择一个元素作为重复的数
3.剩下n-3个数按照顺序可以插入左边或者右边 pow(2, n-3)种
所以结果为C(m,n-1)×(n-2)×pow(2, n-3)
快速幂+卢卡斯

附ac代码:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll mod = 998244353;

long long binpow(long long a, long long b, long long m) {
  a %= m;
  long long res = 1;
  while (b > 0) {
    if (b & 1) res = res * a % m;
    a = a * a % m;
    b >>= 1;
  }
  return res;
}

ll C(ll n, ll m)
{
    if(m > n) return 0;
    ll ans = 1;
    for(int i=1; i<=m; i++)
    {
        ll a = (n + i - m) % mod;
        ll b = i % mod;
        ans = ans * (a * binpow(b, mod-2, mod) % mod) % mod;
    }
    return ans;
}

long long Lucas(long long n, long long m, long long p) {
  if (m == 0) return 1;
  return (C(n % p, m % p) * Lucas(n / p, m / p, p)) % p;
}



int main(int argc, char const *argv[])
{
	ll n, m, ans = 0;
	cin >> n >> m;
	if (n == 2);
	else
	{
		ans = (Lucas(m, n - 1, mod) % mod);
		ans = ans * (n - 2) % mod;
		ans = ans * binpow(2, n - 3, mod) % mod;
	} 

	cout << ans << endl;
	return 0;
}