[C++]背包问题(1):洛谷 采药 01背包模型详解
-01背包拓维- [原题]榨取kkksc03
[输入样例]6 10 10 1 1 2 3 3 2 2 5 5 2 4 3[输出样例]
4[解题思路]
不难看出,这道题是01背包模型的拓展。在上一次的题解中,我们要求从M个物品中选择T时间内最大的价值组合,而∑(items[k].time)<=T则是模型的最大值约束条件(k表示每一个被选择的物品)。
而这道题最大的不同之处在于,存在两个最大值约束条件M和T(注意此处的M与上一题的M含义不同),而且他们的地位是相同的。因此,我们可以用相同的方式去处理他们。
对于单一的约束条件T,我们有
dp[j] = max(dp[j], dp[j - item[i].t] + item[i].val)
那么,对于两个约束条件T和M(显然,由于条件的增多,我们的dp数组也需要对应地升维),我们可以类比得到
dp[i][j] = max(dp[i][j], dp[j - item[i].t][j - item[i].m] + item[i].val)
当然,这道题中并没有给每一个愿望赋予特殊的价值,只需要统计对应的个数,因此这里的item[i].val即为1。
接下来,让我们看一下整道题的代码:
#include-特殊的约束条件:倍数- [原题] [输入样例]#include using namespace std; class Wish { public: int money; int time; Wish(int m = 0,int t = 0):money(m),time(t){} }; int main() { int n, M, T; cin >> n >> M >> T; vector arr(n); for (int i = 0; i < n; i++) { int m, t; cin >> m >> t; arr[i] = Wish(m, t); } vector > dp(M + 1, vector (T + 1, 0)); for (int i = 0; i < n; i++) { for (int j = M; j >= arr[i].money; j--) { for (int k = T; k>= arr[i].time; k--) { dp[j][k] = max(dp[j][k], dp[j - arr[i].money][k - arr[i].time] + 1); } } } cout << dp.back().back(); return 0; }
4 5 1 2 8 2[输出样例]
3[解题思路]
与常规的01背包模型不同的是,这道题给出的约束条件并非最大值,而是要求和为F倍数的方案个数。而倍数问题,我们自然而然就会想到取余。
接下来,我们就通过列表的方式来研究一下余数(我们用j来表示)和物品组合(我们用i来表示)之间,能否找到一个合适的状态转移方程。(以下数据皆取自题目样例)
我们首先建立如下表格,并完成第一行的填写:
接下来是第二行:
我们来对这一行产生的三个1作一下简单的分析:
(1)对于j==1处的1,显然是由上一行的结果直接继承下来的;
(2)对于j==2处的1,则是由本身2%F得到余数2产生的可行方案;
(3)而对于j==3处,则是由于(1+2)%F得到余数3产生的方案,但是我们仍然难以找到一个固定的式子去表示这种关系,在这里,我们用红色将其标注了出来。
对于第三、第四行,我们仍然采用这样的方式去完成,为了更清楚地反映结果,我们用黑色、绿色、红色分别将上述(1)、(2)、(3)相同方式产生的方案标注出来,结果如下:
显然,每一格的结果都是由三个部分决定的,而这三个部分恰好是上述(1)(2)(3)对应的三种情形。由于情形(3)比较复杂,我们这里将其单独拿出来讨论一下。
对于第二行的情形(3)我们是如何得到的?我们由等式(1+2)%F==j(此处F=5,j=3)作出了“此方案可行”的判断。当然,我们可以对这个等式进行适当的变形,即:
1==(j-2+F)%F(先加上F再取余是为了防止j-2为负数)
如果这是个重要的关系式,那么等式左边的1又该如何解释呢?让我们将目光移向上一行:j=1处的值恰好就是转移到这一格的值。那么,这第三部分的值我们也可以用通式表示出来了:
V = dp[i - 1][(j - arr[i] + F) % F]
最终,我们可以得到这样一个状态转移方程:
dp[i][j] = (dp[i - 1][j]) + (arr[i] == j ? 1 : 0) + (dp[i - 1][(j - arr[i] + F) % F])
为了防止数字越界,优化存储空间,我们还可以进行一些处理: *一个简单的预处理由于我们只对n%F的余数进行操作,因此我们可以在数字读取阶段就将每一项%F后储存。
*对每一项计算都执行取模运算由于题目要求结果对10^8取模,我们可以让dp的每一项都对10^8取模后再相加,因为(a+c)%b==(a%b+c%b)%b是个恒等式。
*滚动数组由于dp的每一项结果只与本行和上一行结果有关,因此我们可以将二维数组优化为两个一维数组,具体原理已经在上一个题解中详细解释过。
最后,让我们直接来看代码吧:
#include#include using namespace std; int main() { //数据读取及数组声明 const int mod = 1e8; int N, F; cin >> N >> F; vector arr(N); for (int i = 0; i < N; i++) { int n; cin >> n; arr[i] = n % F;//直接取余后存储 } vector first(F, 0); vector second(F, 0); for (int i = 0; i < N; i++) { first = second;//将second的数据覆盖first的无用数据 for (int j = 0; j < F; j++) { second[j] = (((arr[i] == j ? 1 : 0) + (first[j])) % mod + (first[(j - arr[i] + F) % F]) % mod) % mod;//对每一个都%mod } } //注意我们要求的是“倍数”,因此应返回取余后余数为0的结果 cout << second[0]; return 0; }


![[C++]背包问题(2):洛谷 榨取kkksc03&&Cow Frisbee Team S 更多约束条件下的01背包 [C++]背包问题(2):洛谷 榨取kkksc03&&Cow Frisbee Team S 更多约束条件下的01背包](http://www.mshxw.com/aiimages/31/869207.png)
