BZOJ5369-[Pkusc2018]最大前缀和

5369: [Pkusc2018]最大前缀和

Time Limit: 20 Sec   Memory Limit: 512 MB

Description

小C是一个算法竞赛爱好者，有一天小C遇到了一个非常难的问题：求一个序列的最大子段和。

但是小C并不会做这个题，于是小C决定把序列随机打乱，然后取序列的最大前缀和作为答案。

小C是一个非常有自知之明的人，他知道自己的算法完全不对，所以并不关心正确率，他只关心求出的解的期望值，

现在请你帮他解决这个问题，由于答案可能非常复杂，所以你只需要输出答案乘上n!后对998244353取模的值，显然这是个整数。

注：最大前缀和的定义：i∈[1,n],Sigma(a _j)的最大值,其中1<=j<=i

Input

第一行一个正整数nnn，表示序列长度。

第二行n个数，表示原序列a[1..n]，第i个数表示a[i]。

1≤n≤20,Sigma(|Ai|)<=10^9,其中1<=i<=N

Output

输出一个非负整数，表示答案。

Sample Input

2
-1 2

Sample Output

3

考虑一个分界点p，使得∑A₁~p为最大前缀和，那么显然p之后的所有前缀和均<0，否则就存在可以替换p的方案使得前缀和更大。

用sum[i]表示子集i的数值和，f[i]表示最大前缀和为sum[i]的方案数，从后往前插入，>0就可以转移。g[i]表示任意前缀和均为负的方案数，从前往后插入，<0就可以转移。f[i]和g[i]均可以通过动态规划求得。最后答案即为sum[S]*f[S]*g[S′]。时间复杂度O（2^n*n）

 Code:

#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cstdlib>
#define mod 998244353
#define N 21
using namespace std;
int n,a[N],sum[1<< N],f[1<<N],g[1<<N],num[1<<N];
int lowbit(int x){return x & -x;}
int main()
{
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    {
        scanf("%d",&a[i]);
        num[1<<i]=a[i];
    }
    int maxsta=(1<<n)-1;
    for(int i=0;i<=maxsta;i++)
        sum[i]=sum[i^lowbit(i)]+num[lowbit(i)];
    g[0]=1;
    for(int i=0;i<=maxsta;i++)
        if(sum[i]<=0)
            for(int j=0;j<n;j++)
                if((i>>j)&1)
                    g[i]=(g[i]+g[i^(1<<j)])%mod;
    for(int i=0;i<n;i++)f[1<<i]=1;
    int ans=0;
    for(int i=0;i<=maxsta;i++)
    {
        if(sum[i]>0)
            for(int j=0;j<n;j++)
                if(!((i>>j)&1))
                    f[i|(1<<j)]=(f[i|(1<<j)]+f[i])%mod;
        ans=(ans+1ll*(sum[i]+mod)*f[i]%mod*g[maxsta^i]%mod)%mod;
    }
    printf("%d\n",ans);
    return 0;
}