AtCoder Beginner Contest 217 G - Groups(第二类斯特林数)

传送门

题意：

$n$个整数，为$1$ ~ $n$ ,将其分为 $k$ 组，如果两数模 $m$ 同余，则不能放在一组。求对于 $k\in [1,n]$ ，输出分为 $k$ 组的方案数。

题解：

如果没有限制条件，那么就是普通的第二类斯特林数。

$设dp[i][j]$表示前 $i$ 个数分为 $j$ 组的方案数，转移式为:

$dp[i][j]=dp[i-1][j-1]+j\times dp[i-1][j]$

$dp[i-1][j-1]$表示单独一组，$j\times dp[i-1][j]$ 表示跟前面的数一组，那么就有 $j$ 种选法。

考虑限制条件，其实只会对第二个式子有影响。

设 $x$​ 为前面跟 $i$​ 模$m$​同余的数的个数，那么就只剩下 $j-x$​ 个选法，则转移变为 $dp[i][j]=dp[i-1][j-1]+(j-x)\times dp[i-1][j]$​

代码：

#pragma GCC diagnostic error "-std=c++11"
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <ctime>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <stack>
#define iss ios::sync_with_stdio(false)
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
typedef pair<int, int> pii;
const int mod = 998244353;
const int MAXN = 5e3 + 5;
const int inf = 0x3f3f3f3f;
int num[MAXN];
ll dp[MAXN][MAXN];
int main()
{
    int n, m;
    cin >> n >> m;
    dp[0][0] = 1;
    for (int j = 1; j <= n; j++) {
        for (int i = 0; i <= n; i++) {
            num[i] = 0;
        }
        for (int i = 1; i <= n; i++) {
            dp[i][j] = (dp[i - 1][j - 1] + dp[i - 1][j] * (j - num[i % m]) % mod) % mod;
            num[i % m]++;
        }
    }
    for (int i = 1; i <= n; i++) {
        cout << dp[n][i] << endl;
    }
}