2021百度之星复赛1002第二类斯特林数

题目

题目链接 Add or Multiply 1

题目大意

有一个初始数$x$和一个运算序列$a_{n+m}$，其中$a_1$到$a_n$是加法，$a_{n+1}$到$a_{n+m}$是乘法，令$$S=x+a_1+a_2+...+a_n\times a_{n+1}\times a_{n+2}\times ...\times a_{n+m}$$

两个运算序列被称为是本质相同的，当且仅当这两个序列的$x$和$a_i$取任何值时，得到的$S$都相同。

给出$n$和$m$，可以重新排列运算序列$a$，问有多少个本质不同的运算序列。运算序列$a$从左到右执行。

解题思路

我们将$+a_i$记作$0_i$，$\times a_i$记作$1_i$。最后的运算序列即可表示为一个$01$序列。

设这个$01$序列为
$0_1{0}_2{0}_3{1}_4{1}_5{1}_6{0}_7{0}_8{0}_9$

对于任意的$x$和$a_i$取值，两个序列得到的值都要相同，那么就要满足对于一个连续相同的子段，包含的元素的个数和值必须相同。对于上述的这个序列，以下的这些$01$序列都和它本质相同：

$0_3{0}_2{0}_1{1}_4{1}_5{1}_6{0}_7{0}_8{0}_9$
$0_3{0}_1{0}_2{1}_4{1}_5{1}_6{0}_7{0}_8{0}_9$
$0_1{0}_2{0}_3{1}_6{1}_4{1}_5{0}_7{0}_8{0}_9$
$......$

而
情况1：$0_7{0}_8{0}_9{1}_4{1}_5{1}_6{0}_1{0}_2{0}_3$
情况2：$0_1{0}_2{0}_7{1}_4{1}_5{1}_6{0}_1{0}_8{0}_9$
$......$
则本质不同。

让我们先仅考虑其中一种运算，也即$0$序列的排列方案数。

设有分成的$k$段连续$0$序列，这k段之间可以重排列，这样获得的序列一定本质不同，也即情况$1$，方案数为$k!$ 。每一段序列中的元素不同，序列也一定本质不同，也即情况$2$，要将$n$个不同的数分成$k$个集合，方案数为第二类斯特林数。

将n个相同的物品放进k个相同的集合中，方案数可以用隔板法求。
将n个不相同的物品放进k个相同的集合中，方案数可以用第二类斯特林数求。
将n个不相同的物品放进k个不相同的集合中，方案数可以用第二类斯特林数求，只需对集合进行重排列，也即乘上$k!$。

本题就是考察第三个知识点。

代码

#include <bits/stdc++.h>
using namespace std;

const int N = 3e3 + 5;
const long long mod = 1e9 + 7;

long long fact[N];
void init_fact()
{
    fact[0] = 1;
    for (int i = 1; i < N; i++)
        fact[i] = fact[i - 1] * i % mod;
}

long long n, k;
long long S[N][N];
void init_S()
{
    for (int i = 1; i < N; i++)
        S[i][0] = 0, S[i][i] = S[i][1] = 1;
    for (int i = 2; i < N; i++)
        for (int j = 1; j < i; j++)
            S[i][j] = (S[i - 1][j - 1] + 1ll * j * S[i - 1][j] % mod) % mod;
}
void solve()
{
    scanf("%lld %lld", &n, &k);
    if (n < k)
        swap(n, k);
    long long ans = 0;
    for (long long i = 1; i <= k; i++)
    {
        //最左边放一个隔板 中间将n个数分成i个集合的方案数 （最右边同理  方案数*2
        ans = (ans + 2 * S[n][i] % mod * fact[i] % mod * S[k][i] % mod * fact[i] % mod) % mod;

        //两边都放一个数  中间将n个数分成i-1个集合的方案数
        if (i >= 2)
            ans = (ans + S[n][i - 1] * fact[i - 1] % mod * S[k][i] % mod * fact[i] % mod) % mod;

        //两边都不放  中间将n个数分成i+1个集合的方案数
        if (i + 1 <= n)
            ans = (ans + S[n][i + 1] * fact[i + 1] % mod * S[k][i] % mod * fact[i] % mod) % mod;
    }
    printf("%lld\n", ans);
}
int main()
{
    init_fact();
    init_S();
    int t;
    scanf("%d", &t);
    while (t--)
        solve();
    return 0;
}