牛客 每日一题 失衡天平 dp

失衡天平

题目链接

题意

给一个数组（代表一个东西的重量），每次选一些东西出来分成两堆，两堆的重量的差不能超过m，问最多可以获得的重量是多少。
（可以选多次，但一个东西只能选一次）

题解。

被他可以选多次忽悠了。
例如样例给的：
5 4
1 5 61 65 100
第一次选 1 5
第二次选 61 65
答案就是132.
但是选多次跟选一次是一样的。因为 可以选1 + 65 5 + 61.
这样的话，就dp
dp[i][j]代表 前i个分成两组，两边的差的绝对值为j的最大重量和。
于是遍历到当前i点的时候ai可以放到左边也可以放到右边。
所以状态转移：

dp[i][j + a[i]] = max(dp[i][j + a[i]],dp[i - 1][j] + a[i]);
dp[i][j - a[i]] = max(dp[i][j - a[i]],dp[i - 1][j] + a[i]);

代码:

#include <algorithm>
#include <cstdio>
#include <iostream>
#include <vector>
#include <stack>
#include <queue>
#include <map>
#include <cmath>
#include <set>
#include <cstring>
#include <string>