题目描述
思路
需要从n颗雨花石中选出一些,使得这些雨花石的重量之和等于所有雨花石重量的一半,从而能够平分给两个人。但是,题目希望我们使用的雨花石数量最少。这跟我们熟知的01背包问题有点相似。
既然所有雨花石的重量和是sum,那么每人应得的雨花石重量就是sum/2,我们称之为“背包承重”。这样,题目转换成了:如何从雨花石中选出一些,使得它们的重量和是sum/2,但是选的雨花石数量尽量少?
使用一个二维数组dp,其中dp[i][j]的意思是,从前i颗雨花石中,如何选出一些使得它们的重量和为j,并且使得选的雨花石数量尽量少?
状态转移方程:一种是考虑第i颗雨花石,一种是不考虑第i颗雨花石。所以,可以用以下方式来更新dp[i][j]的值:
如果选第i颗雨花石,那么dp[i][j] = dp[i-1][j-nums[i-1]] + 1(即我们要加上这一颗雨花石的重量和数量)。
如果不选第i颗雨花石,那么dp[i][j] = dp[i-1][j](即维持原样)。
选择这两种情况中的较小值作为dp[i][j]。
代码
def melon_distribution<