思路:01背包
为什么能转化为01背包问题呢?
题目转化为:将所有石头分成两堆,这两堆的重量,尽可能相近,减掉得到的abs才会尽量小,也就是sum/2的背包最多能放多少重量的石头 a, sum-a-a就是ans。
一次碰撞就是一正一负 ,计算出的结果再跟其他一正一负的过程,每个元素之前构造+ /- 形成的表达式得到的最小正整数就是石头的最小重量,就转化为背包容量为sum/2最多能放多少重量的石头,每个石头只能选一次,即0 - 1背包问题
dp[i][j]表示容量为j的背包,放前i个元素最多能放多少重量的石头
转移方程:dp[i][j]=max(dp[i-1][j],dp[i-1][j-stones[i]] + stones[i])
dp[n-1][sum/2] 为背包容量为sum/2最多能放多少重量的石头
ans:abs(sum - dp[sum / 2] - dp[sum / 2])
class Solution {
public:
int lastStoneWeightII(vector<int>& stones) {
int n = stones.size();
int sum = accumulate(stones.begin(), stones.end(), 0);
vector<int> dp(sum / 2 + 1);
for (int i = 0; i < n; ++i) {
for (int j = sum / 2; j >= stones[i]; --j) {
dp[j] = max(dp[j], dp[j - stones[i]] + stones[i]);
}
}
return abs(sum - dp[sum / 2] - dp[sum / 2]);
}
};
易错点:看到30以为dfs?刚开始太天真了,每次枚举两个时个数为n(n-1)/2,复杂度比n!还要大,所以肯定炸。