洛谷 P5369
先规定一些东西:
若存在多个 p 使得 ∑pi=1ai 最大,默认最大(即最靠右)的一个 p 是它的最大前缀和的位置。
U 表示全集。
假设 p 是最大前缀和的位置,则说明不存在 1<x≤p 满足 ∑pi=xai<0,否则去掉这段答案不会变小。并且不存在 p<x≤n 满足 ∑xi=p+1ai≥0,否则我们把这一段加上之后不会更劣。
而且数据范围那么小就考虑状压。
令 sumi 表示集合 i 内所有数的和,fi 表示集合 i 内的数组成的排列,最大前缀和 =sumi 的方案数,gi 表示集合 i 内的数组成的排列,所有的最大前缀和都 <0 的方案数。
显然最终答案为 ∑isumi×fi×gU−i
先考虑如何求出 gi。
当 sumi≥0 时,gi=0。否则 gi=∑j∈igi−j。
然后 fi 的话考虑刷表,在集合 i 的排列前加入一个数 j(j∉i)。
如果 sumi≥0,那么 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;
}
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
· 25岁的心里话