leetcode 1049. Last Stone Weight II

最新推荐文章于 2024-08-14 13:17:23 发布

原创最新推荐文章于 2024-08-14 13:17:23 发布 · 1.8k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#DP

LeetCode 专栏收录该内容

66 篇文章

订阅专栏

博客围绕LeetCode 1049题展开，该题要求挑任意两块石头相撞，求最终剩下的最小值。解题思路是将数组分成两个堆，用动态规划（DP）求解。dp[i]表示能否组成重量为i的堆，另一个堆为sum - i，当i与sum/2最接近时，两堆差值最小。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

leetcode 1049. Last Stone Weight II

题意:与第一题不同，这次是挑任意两块相撞，求最终剩下的最小值。

思路：可以看成将当前数组分成两个堆，两堆里面的石头分别对撞。也就是求两个堆的石头后，然后做差，求最小。可以用dp来做。dp[i]表示能否组成重量为i的堆，那么另外一个堆就是sum-i。当i与sum/2最接近时，两个差最小。

代码：

class Solution {
public:
    int lastStoneWeightII(vector<int>& stones) {
        int sum = 0;
		vector<int> dp(1600, 0);
        dp[0]=1;
		for (auto stone : stones)
		{
			sum += stone;
			for (int i = 1500; i - stone >= 0; i--)
			if (dp[i - stone])
				dp[i] = 1;
		}

		for (int i = sum/2 ; i >= 0; i--)
			if (dp[i])
                return (sum - i) - i;
		return 0;
    }
};