PKUSC2018 最大前缀和

题意

数据范围

Analysis

题目事实上就是让我们求，给出的数的全排列的最大前缀和的和。考虑怎么做？首先分析一下性质，满足最大前缀和的位置，假设在$p$。那么在$p$之后，不能找到一个位置$x$,使得$\sum_{i=p+1}^{x} a_i > 0$，在$p$之前，不能够找到一个位置$x$，使得，$\sum_{i=x}^pa_i <= 0$，那我们就可以构造出这样的排列来得到答案，这个用状压$DP$解决，设状态$f_i$表示用了状态$i$的数，构造出一个排列，使得排列的后缀和都大于$0$，$g_i$跟$f_i$类似，使排列的前缀和都小于等于$0$，$sum_i$表示$i$状态下所有数的和，那么$Ans=\sum_if_i*sum_i*g_{(1<<n)-1-i}$。

Code

# include<cstdio>
# include<cstring>
# include<algorithm>
using namespace std;
const int N = (1 << 20) + 5;
const int mo = 998244353;
typedef long long ll;
int f[N],g[N],sum[N],a[25];
int n,ans,lim;
int main()
{
    scanf("%d",&n); lim = (1 << n) - 1;
    for (int i = 1 ; i <= n ; ++i) scanf("%d",&a[i]);
    for (int i = 1 ; i <= lim ; ++i)
        for (int j = 0 ; j < n ; ++j)
            if (i & (1 << j)) { sum[i] = sum[i ^ (1 << j)] + a[j + 1]; break; }
    g[0] = 1;
    for (int i = 1 ; i <= lim ; ++i)
    {
        if (sum[i] <= 0)
            for (int j = 0 ; j < n ; ++j)
                if ((i & (1 << j))) g[i] = (g[i] + g[i ^ (1 << j)]) % mo;
    }
    for (int i = 0 ; i < n ; ++i) f[1 << i] = 1;
    for (int i = 1 ; i <= lim ; ++i)
    {
        if (sum[i] > 0)
            for (int j = 0 ; j < n ; ++j)
                if (!(i & (1 << j))) f[i | (1 << j)] = (f[i | (1 << j)] + f[i]) % mo;
        ans = (ans + (ll)f[i] * sum[i] % mo * g[lim - i] % mo) % mo;
    }
    printf("%d\n",(ans + mo) % mo);
    return 0;
}