洛谷 P5369

先规定一些东西:

若存在多个 p 使得 pi=1ai 最大,默认最大(即最靠右)的一个 p 是它的最大前缀和的位置。

U 表示全集。


假设 p 是最大前缀和的位置,则说明不存在 1<xp 满足 pi=xai<0,否则去掉这段答案不会变小。并且不存在 p<xn 满足 xi=p+1ai0,否则我们把这一段加上之后不会更劣。

而且数据范围那么小就考虑状压。

sumi 表示集合 i 内所有数的和,fi 表示集合 i 内的数组成的排列,最大前缀和 =sumi 的方案数,gi 表示集合 i 内的数组成的排列,所有的最大前缀和都 <0 的方案数。

显然最终答案为 isumi×fi×gUi

先考虑如何求出 gi

sumi0 时,gi=0。否则 gi=jigij

然后 fi 的话考虑刷表,在集合 i 的排列前加入一个数 jji)。

如果 sumi0,那么 fi+j+=fi,否则 fi 没有贡献。

时间复杂度 O(n2n)

Code:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 20, M = 1 << N, mod = 998244353;
int n, m;
int a[N];
ll sum[M]; int f[M], g[M];

void add(int &a, int b) {
	a += b;
	if (a >= mod) a -= mod;
	if (a < 0) a += mod;
}

int main() {
	scanf("%d", &n), m = 1 << n;
	for (int i = 0; i < n; ++i) scanf("%d", &a[i]), sum[1 << i] = a[i], f[1 << i] = 1;
	for (int i = 1; i < m; ++i) sum[i] = sum[i ^ (i & -i)] + sum[i & -i];
	g[0] = 1;
	for (int i = 1; i < m; ++i) {
		if (sum[i] >= 0) {
			for (int j = 0; j < n; ++j) if (!(i >> j & 1)) add(f[i | (1 << j)], f[i]);
		}
		else {
			for (int j = 0; j < n; ++j) if (i >> j & 1) add(g[i], g[i ^ (1 << j)]);
		}
	}
	int ans = 0;
	for (int i = 1; i < m; ++i) add(ans, sum[i] * f[i] % mod * g[(m - 1) ^ i] % mod);
	printf("%d", ans);
	return 0;
}
posted @   Kobe303  阅读(20)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
· 25岁的心里话
点击右上角即可分享
微信分享提示