首页 > 其他分享 >ABC221H

ABC221H

时间:2022-10-23 06:33:05浏览次数:40  
标签:int sum 直方图 ABC221H 移动 mod

题意:对于每个 \(k=1\sim n\),统计满足以下要求的无序集合个数:

  • 元素均为正整数。
  • 恰有 \(k\) 个元素。
  • 同一个数最多出现 \(m\) 次。
  • 元素和为 \(n\)。

答案对 \(998244353\) 取模。

\(m\leq n\leq 5000\)。

对于一种方案,将集合中的数从大到小排列,画出直方图。

直方图的上边缘是一条路径,要求如下:

  • 从左侧某处出发
  • 单调不升
  • 围成面积为 \(n\)
  • 最多横向连续移动 \(m\) (这个横过来不好做)
  • 恰横向移动了 \(k\)

横过来看,发现路径变成了以下这样:

  • 从左侧下方某处出发
  • 围成面积为 \(n\)
  • 最多纵向连续移动 \(m\)
  • 恰纵向移动了 \(k\)

考虑 DP,记 \(f_{i,j}\) 表示当前面积为 \(i\),高度为 \(j\) 的方案数。

枚举上一行的高度,转移如下:

\[f_{i,j}=\sum\limits_{k=\max(1,j-m)}^{\min(j,i-j)}f_{i-j,k} \]

初值为 \(f_{0,0}=1\),最终就是输出 \(f_{n,*}\)。

前缀和优化,时间复杂度 \(\mathcal O(n^2)\)。

Code:

#include <bits/stdc++.h>
using namespace std;
const int N = 5005, mod = 998244353;
int n, m;
int f[N][N], sum[N][N];

int main() {
	scanf("%d%d", &n, &m);
	sum[0][0] = 1;
	for (int i = 1; i <= n; ++i)
		for (int j = 1; j <= i; ++j) {
			f[i][j] = sum[i - j][min(j, i - j)];
			if (j > m) (f[i][j] += mod - sum[i - j][min(j - m - 1, i - j)]) %= mod;
			sum[i][j] = (sum[i][j - 1] + f[i][j]) % mod;
		}
	for (int i = 1; i <= n; ++i) printf("%d\n", f[n][i]);
	return 0;
}

标签:int,sum,直方图,ABC221H,移动,mod
From: https://www.cnblogs.com/Kobe303/p/16817821.html

相关文章