正題

題目鏈接:https://www.luogu.com.cn/problem/P5369

---

題目大意

一個數列$a$的權值定義為$max\{{\sum }_{i=1}^k{a}_i\}(k\in [1,n])$
給出$n$個數字，求它們所有排列的權值和

$1\le n\le 20$

---

解題思路

設$s_i,f_i,g_i$分別表示集合$i$的權值和，集合$i$的所有排列中最大前綴和為$s_i$的方案數，集合$i$的所有排列中的最大前綴和為負的方案數。那么答案就是
$$\sum _{i=0}^{2^n?1}{f}_i{s}_i{g}_{2^n?1?i}$$
$s_i$很好求。$g_i$的話我們只轉移$s_i<0$的就可以了，$f_i$的話我們考慮每次在前面插入一個數，那么只要原來的是最大前綴和，那么插入之后也一定是。

時間復雜度$O(2^{n}n)$

---

code

#include<cstdio> #include<cstring> #include<algorithm> using namespace std; const int N=21,P=998244353; int n,a[N],lg[1<<N],s[1<<N],f[1<<N],g[1<<N],ans; int main() {scanf("%d",&n);for(int i=0;i<n;i++)scanf("%d",&a[i]);for(int i=0;i<n;i++)lg[1<<i]=i;int MS=(1<<n);f[0]=g[0]=1;for(int i=1;i<MS;i++){int p=i&-i;s[i]=(s[i-p]+a[lg[p]])%P;}for(int i=0;i<MS;i++){if(s[i]<0)continue;for(int j=0;j<n;j++){if(i&(1<<j))continue;(f[i|(1<<j)]+=f[i])%=P;}}for(int i=0;i<MS;i++){for(int j=0;j<n;j++){if(i&(1<<j))continue;int z=i|(1<<j);if(s[z]<0)(g[z]+=g[i])%=P;}}for(int i=0;i<MS;i++)(ans+=1ll*f[i]*g[MS-1-i]%P*s[i]%P)%=P;printf("%d\n",(ans+P)%P);return 0; }

總結

以上是生活随笔為你收集整理的P5369-[PKUSC2018]最大前缀和【状压dp】的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。