最后一块石头的重量 II

题在确定思路后要反复验证,代入各种情况去试试,这道题我认为就是一个贪心的题,并且代入两种情况过了就直接写了。但是到最后有差不多10%用例通不过,思路完全错误,这是道DP的题。
题目链接++++++++++++++++++++++++++++++
题目为:
有一堆石头,用整数数组 stones 表示。其中 stones[i] 表示第 i 块石头的重量。

每一回合,从中选出任意两块石头,然后将它们一起粉碎。假设石头的重量分别为 x 和 y,且 x <= y。那么粉碎的可能结果如下:

如果 x == y,那么两块石头都会被完全粉碎;
如果 x != y,那么重量为 x 的石头将会完全粉碎,而重量为 y 的石头新重量为 y-x。
最后,最多只会剩下一块 石头。返回此石头 最小的可能重量 。如果没有石头剩下,就返回 0。

示例 1:

输入:stones = [2,7,4,1,8,1]
输出:1
解释:
组合 2 和 4,得到 2,所以数组转化为 [2,7,1,8,1],
组合 7 和 8,得到 1,所以数组转化为 [2,1,1,1],
组合 2 和 1,得到 1,所以数组转化为 [1,1,1],
组合 1 和 1,得到 0,所以数组转化为 [1],这就是最优值。
示例 2:

输入:stones = [31,26,33,21,40]
输出:5
示例 3:

输入:stones = [1,2]
输出:1

1 <= stones.length <= 30
1 <= stones[i] <= 100

解析:
当时一看题意就觉得一排序然后将相邻两个相减再放入就行了,但是在数组大小为奇数的情况下过不了。这个用例里的数据为奇数个并且数值间的差异并不大,这样我计算到最后肯定是一个大数减一个极小的数,刚开始选择用这个思路时应该把各种情况考虑清楚,我的思路仅对偶数个数和数组里全是相同的数成立。
错误代码

class Solution {
public:
    int lastStoneWeightII(vector<int>& stones) {
        //贪心
        priority_queue<int> q;
        for (int i=0;i<stones.size();i++){
            q.push(stones[i]);
        }
        while (q.size()>1){
            int x = q.top();
            q.pop();
            int y = q.top();
            q.pop();
            q.push(x-y);
        }
        if (q.size()==1) return q.top();
        else
        return 0;
    }
};

这时就应该再次分析了;
一看题干十分像石子合并区间dp,但是这道题是让把两石子的差值再次放入数组,然后重新计算。相当于有两堆石子,求把所有石子放入两堆石子中使其差值的绝对值最小。设两堆石子总数分别为x,y,设x为较多的那一堆。
x+y = sum;
求x-y就等价于转换为sum-2y。仔细一想就是让求在所有的石子中求出可组合的小于sum/2的最大值,这是让求最大值就想到了DP。
正确代码

class Solution {
public:
    int lastStoneWeightII(vector<int>& ss) {
 int n = ss.size();
        int sum = 0;
        for (int i : ss) sum += i;
        int t = sum / 2;
        vector<vector<int>> f(n + 1, vector<int>(t + 1));
        for (int i = 1; i <= n; i++) {
            int x = ss[i - 1];
            for (int j = 0; j <= t; j++) {
                f[i][j] = f[i - 1][j];
                if (j >= x) f[i][j] = max(f[i][j], f[i - 1][j - x] + x);
            }
        }
        return sum - f[n][t] - f[n][t];
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值