mirror of
https://github.com/heqin-zhu/algorithm.git
synced 2024-03-22 13:30:46 +08:00
33 lines
1.2 KiB
Python
33 lines
1.2 KiB
Python
#coding: utf-8
|
||
''' mbinary
|
||
#######################################################################
|
||
# File : last-stone-weight.py
|
||
# Author: mbinary
|
||
# Mail: zhuheqin1@gmail.com
|
||
# Blog: https://mbinary.xyz
|
||
# Github: https://github.com/mbinary
|
||
# Created Time: 2019-05-28 23:30
|
||
# Description:
|
||
leetcode 1049: https://leetcode-cn.com/problems/last-stone-weight-ii/
|
||
|
||
有一堆石头，每块石头的重量都是正整数。
|
||
|
||
每一回合，从中选出任意两块石头，然后将它们一起粉碎。假设石头的重量分别为 x 和 y，且 x <= y。那么粉碎的可能结果如下：
|
||
|
||
如果 x == y，那么两块石头都会被完全粉碎；
|
||
如果 x != y，那么重量为 x 的石头将会完全粉碎，而重量为 y 的石头新重量为 y-x。
|
||
最后，最多只会剩下一块石头。返回此石头最小的可能重量。如果没有石头剩下，就返回 0。
|
||
|
||
#######################################################################
|
||
'''
|
||
|
||
class Solution:
|
||
def lastStoneWeightII(self, stones: List[int]) -> int:
|
||
sm = sum(stones)
|
||
ans = sm//2
|
||
dp = [0]*(ans+1)
|
||
for x in stones:
|
||
for j in range(ans,x-1,-1):
|
||
dp[j] = max(dp[j],dp[j-x]+x)
|
||
return sm-2*dp[ans]
|