18牛客多校第六场(排列组合)-CSDN博客

本文链接：https://blog.csdn.net/weixin_45691711/article/details/120633633

题目大意：

有 $N(\le 1e18)$ 个空集合（集合内元素不可重复），被依次标记为 $\sim N$ ，会执行 $N$ 次操作，第 $i$ 次操作，从 $1\sim M(\le 1e18)$ 中选择一个数 $x$ ，在 $i\sim N$ 的集合中加入元素 $x$ 。执行完 $N$ 次操作之后，有多少可能不同的结果，答案对 $998244353$ 取模

$\le 1e6$

解题思路：（思路来源：lkaiii）

首先第一种操作有 $M$ 种填法，而之后的 $2\sim N$ ，倘若再填第一种操作的数，就相当于没填
设还要放 $k$ 种数，那么就相当于从 $n - 1$ 个箱子中选出 $k$ 个来放这 $k$ 种数，即 $C_{n-1}^k$ ，而这 $k$ 种数要从 $m - 1$ 中取且其有顺序，即 $A_{m-1}^k$ 种，所以答案是： $m\sum_{k=0}^{min(n-1,m-1)}C_{n-1}^kA_{m-1}^k$
对于剩下的 $n - 1 - k$ 个操作，可以都填第一种操作数，所以就相当于是没填了，只管这 $k$ 个就行

AC代码：

#include <bits/stdc++.h>
#define ft first
#define sd second
#define pb push_back
#define IOS ios::sync_with_stdio(false), cin.tie(0), cout.tie(0) //不能跟puts混用
#define seteps(N) fixed << setprecision(N)
#define endl "\n"
const int maxn = 1e5 + 10;
using namespace std;
typedef long long ll;
typedef double db;
typedef pair<int, int> pii;
const ll mod = 998244353;
ll qpow(ll a, ll b) {
   ll res = 1;
   while (b) {
       if (b & 1) res = res * a % mod;
       a = a * a % mod;
       b >>= 1;
   }
   return res;
}
int t, cas = 0;
ll n, m;
int main() {
    IOS;
    cin >> t;
    while (t--) {
        cin >> n >> m;
        ll c = 1, a = 1, ans = 0, rc = n - 1, ra = m - 1;
        for (ll k = 0; k <= min(n - 1, m - 1); k++) {
            ans = (ans + m % mod * c % mod * a % mod) % mod;
            c = c * qpow(k + 1, mod - 2) % mod * (rc % mod) % mod;
            rc--;
            a = a * (ra % mod) % mod;
            ra--;
        }
        cout << "Case #" << ++cas << ": " << ans << endl;
    }
    return 0;
}