[AGC035F]Two Histograms

Description

你有一个 \(N\) 行、\(M\) 列的、每个格子都填写着 0 的表格。你进行了下面的操作:

• 对于每一行 \(i\) ,选定自然数 \(r_i\ (0 ≤ r i ≤ M )\)
  ,将这一行最左边的 \(r_i\) 个格子中
  的数 \(+1\).

• 对于每一列 \(i\) ,选定自然数 \(c_i\ (0 ≤ c i ≤ N )\)
  ,将这一列最上边的 \(c_i\) 个格子中
  的数 \(+1\).

这样,根据你选定的 \(r_1 , r_2 , . . . , r_N , c_1 , c_2 , . . . , c_M\) ,你就得到了一个每个格子要 么是 \(0\) ,要么是 \(1\) ,要么是 \(2\) 的一个最终的表格。

问本质不同的最终表格有多少种。 两个表格本质不同当且进当它们有一个对应格子中的数不同。

\(n, m\le 5\times 10^5\)

Solution

csy的题解：

组合数那里上下写反了，乘上 \(k!\) 是为了让选出来的k行k列一一匹配上，可以固定一个，另一个排列，就是 \(k!\).

Code

#include <iostream>
#include <cstdio>

#define LL long long

using namespace std;

const int maxN = (int) 5e5;
const int mod = (int) 998244353;

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 n, m, ans;
int fac[maxN + 2], ifac[maxN + 2];

void init(int n)
{
    fac[0] = 1;
    for (int i = 1; i <= n; ++i) fac[i] = 1ll * fac[i - 1] * i % mod;
    ifac[n] = qpow(fac[n], mod - 2);
    for (int i = n - 1; i >= 0; --i) ifac[i] = 1ll * ifac[i + 1] * (i + 1) % mod;
}

int C(int n, int m)
{
    if (n < 0 || m < 0 || m > n) return 0;
    return 1ll * fac[n] * ifac[m] % mod * ifac[n - m] % mod;
}

int main()
{
#ifndef ONLINE_JUDGE
    freopen("AGC035F.in", "r", stdin);
    freopen("AGC035F.out", "w", stdout);
#endif

    scanf("%d%d", &n, &m);
    if (n < m) swap(n, m);
    init(n);
    for (int k = 0; k <= m; ++k)
    {
        ans += 1ll * qpow(-1, k) * C(n, k) % mod * C(m, k) % mod * fac[k] % mod * qpow(m + 1, n - k) % mod * qpow(n + 1, m - k) % mod;
        ans %= mod;
        (ans += mod) %= mod;
    }
    printf("%d\n", (1ll * ans + mod) % mod);
}

转载于:https://www.cnblogs.com/cnyali-Tea/p/11446695.html