[CF1442 D] Sum(分治优化dp + 结论)
CF1442D Sum
- problem
- solution
- code
problem
luogu翻譯
solution
部分分做法,預處理每組前綴和,暴力背包 dpdpdp 轉移:dpi,j=max?{dpi?1,j?k+sumi(k)∣0≤k≤l[i]}dp_{i,j}=\max\Big\{dp_{i-1,j-k}+sum_i(k)\ \Big|\ 0\le k\le l[i]\Big\}dpi,j?=max{dpi?1,j?k?+sumi?(k)?∣∣∣??0≤k≤l[i]}
正解也是背包,只不過需要發現一個性質。
必然存在一種最優方案滿足只有最多一組的產品是只抽了一部分的 。
證明:
若不同的兩組 i,ji,ji,j 都只取了一部分,假設 iii 組取了 xxx 個,jjj 組取了 yyy 個。
由題目性質可得:ai[x]≤ai[x+1]∧aj[y]≤aj[y+1]a_i[x]\le a_i[x+1]\wedge a_j[y]\le a_j[y+1]ai?[x]≤ai?[x+1]∧aj?[y]≤aj?[y+1]。
而 ai[x],aj[y]a_i[x],a_j[y]ai?[x],aj?[y] 之間一定有一個大小關系,所以必然有 ai[x]≤aj[y+1]∨aj[y]≤ai[x+1]a_i[x]\le a_j[y+1]\vee a_j[y]\le a_i[x+1]ai?[x]≤aj?[y+1]∨aj?[y]≤ai?[x+1]。
這里不妨假設 ai[x]≤aj[y+1]a_i[x]\le a_j[y+1]ai?[x]≤aj?[y+1] 成立,則有 ....≤ai[x?1]≤ai[x]≤aj[y+1]≤aj[y+2]≤.......\le a_i[x-1]\le a_i[x]\le a_j[y+1]\le a_j[y+2]\le.......≤ai?[x?1]≤ai?[x]≤aj?[y+1]≤aj?[y+2]≤...,那么將 jjj 組選滿顯然更優,或者說至少不劣于。
所以可以不斷構造出一種不劣于原方案且少取一組的方案。
每次對兩個只取一部分的組進行上述調整過程即可得到上述結論。
只有一組是特殊的,容易想到枚舉這個特殊的組。其余組都是整組整組的轉移。
分治優化背包 dpdpdp 即可。
在每個葉子點表示這組是特殊的,分治下來的途中背包 dpdpdp 出 dpd,j:dp_{d,j}:dpd,j?: 前 ddd 層選了 jjj 個的最大值。
code
#include <bits/stdc++.h> using namespace std; #define int long long #define maxn 3005 int f[20][maxn]; int sum[maxn], len[maxn]; int g[maxn][maxn]; int n, k, ans;void dfs( int d, int l, int r ) {if( l == r ) {ans = max( ans, f[d][k] );int w = 0;for( int i = 1;i <= min( len[l], k );i ++ ) {w += g[l][i];ans = max( ans, f[d][k - i] + w );}return;}int mid = ( l + r ) >> 1;for( int i = 0;i <= k;i ++ )f[d + 1][i] = f[d][i];for( int i = l;i <= mid;i ++ )for( int j = k;j >= len[i];j -- )f[d + 1][j] = max( f[d + 1][j], f[d + 1][j - len[i]] + sum[i] );dfs( d + 1, mid + 1, r );for( int i = 0;i <= k;i ++ )f[d + 1][i] = f[d][i];for( int i = mid + 1;i <= r;i ++ )for( int j = k;j >= len[i];j -- )f[d + 1][j] = max( f[d + 1][j], f[d + 1][j - len[i]] + sum[i] );dfs( d + 1, l, mid ); }signed main() {scanf( "%lld %lld", &n, &k );for( int i = 1;i <= n;i ++ ) {scanf( "%lld", &len[i] );for( int j = 1, x;j <= len[i];j ++ ) {scanf( "%lld", &x );if( j <= k ) g[i][j] = x, sum[i] += x;}}dfs( 1, 1, n ); printf( "%lld\n", ans );return 0; }總結
以上是生活随笔為你收集整理的[CF1442 D] Sum(分治优化dp + 结论)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: (成为ddos攻击中间通道)
- 下一篇: [AtCoder Regular Con