鸡数题：关于第二类斯特林数的运用和代码写法

jtss123

已于 2024-02-17 21:13:00 修改

阅读量374

点赞数 9

分类专栏：牛客文章标签：算法

于 2024-02-17 21:12:28 首次发布

本文链接：https://blog.csdn.net/jtss123/article/details/136141531

版权

牛客专栏收录该内容

1 篇文章 0 订阅

订阅专栏

本文介绍了第二类斯特林数的概念及其递推公式，探讨了如何使用容斥原理解决排列组合问题，并提供了AC代码示例，重点在于处理大数情况下组合数的计算，通过预处理优化了算法效率。

摘要由CSDN通过智能技术生成

第二类斯特林数：

概念：将n个不同元素放入m个相同集合内，每个集合不为空，表示为S（n,m）

公式1（递推）：

S(n,m)=S(n-1,m-1)+m*S(n-1,m)

把n当球，m当箱子，第n个球放入m个箱子时，有两种选择，单独放入第m个箱子内，或者不单独放入m个箱子内，单独放入就是S(n-1,m-1) 不单独放入的方法有m*S(n-1,m)

公式2：

推导过程设计容斥原理，因为我不会所以推不动

题目思路：

把题干读完，整理题干信息可得

我们把数组中m个数以二进制表示，那么只需要该数前面n位不全为0，其他位为0。同时每个数从上到下满足单调递增，且1到n列中每列仅有一个1，那么这题相当于是把n个1中放入m列

我们要不要考虑单调性呢？不需要，因为你要做的只是把n个1按照不同组合放入m集合中，排序自己排好组成一个数组就行了，那么这题答案就是S(n,m)

这题因为数字太大了，不能用递推，递推复杂度是n方，必然TLE，因此要采取直接用公式的作法，那么用公式的做法需要做预处理

部分代码展示:

1.组合数求法

因为已经求出了逆元和阶乘，直接上数组。

ll C(int n, int m) // 求组合数
{
    if (m < 0 || m > n || n < 0)
        return 0;
    return c[n] * invc[m] % mod * invc[n - m] % mod; // 组合数公式 n阶乘除m阶乘除n-m阶乘，这里因为要取mod ，因此逆元处理
}

2.阶乘递推求逆元

n的阶乘逆元为inv[n]，那么可以变形为(n-1)!*n*inv[n] $\equiv$ 1(modp)，那么(n-1)的阶乘的逆元为inv[n]*n，得到逆着的递推式子：

ll inv(ll x)
{
    return ksm(x, mod - 2);
}
void init()
{
    c[0] = 1;
    // 求阶乘
    for (int i = 1; i < N; i++)
        c[i] = c[i - 1] * (i % mod) % mod;
    // 求阶乘最大逆元
    invc[N - 1] = ksm(N - 1, mod - 2);
    for (int i = N - 2; i >= 0; i--)
        invc[i] = invc[i + 1] * (i + 1) % mod;
}

AC代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e5 + 10;
ll res, mod = 1e9 + 7;
ll c[N], invc[N];
int n, m;
ll ksm(int a, int b)
{
    ll temp = 1, base = a;
    while (b)
    {
        if (b & 1)
            temp = temp * base % mod;
        base = base * base % mod;
        b >>= 1;
    }
    return temp;
}
ll inv(ll x)
{
    return ksm(x, mod - 2);
}
void init()
{
    c[0] = 1;
    // 求阶乘
    for (int i = 1; i < N; i++)
        c[i] = c[i - 1] * (i % mod) % mod;
    // 求阶乘最大逆元
    invc[N - 1] = inv(c[N - 1]);
    for (int i = N - 2; i >= 0; i--)
        invc[i] = invc[i + 1] * (i + 1) % mod;
}
ll C(ll a, ll b)
{
    if (a < 0 || b < 0 || b > a)
        return 0;
    else
    {
        return c[a] * invc[b] % mod * invc[a - b] % mod;
    }
}
int main()
{
    cin >> n >> m; // 求S（n,m)
    res = 0;
    init();
    for (int k = 0; k <= m; k++)
    {
        if (k % 2 == 0)
            res = (res + C(m, k) * ksm(m - k, n) % mod) % mod;
        else
            res = (res - C(m, k) * ksm(m - k, n) % mod + mod) % mod;
    }
    cout << res * invc[m] % mod;
    return 0;
}