有一堆石头,每块石头的重量都是正整数。
每一回合,从中选出两块 最重的 石头,然后将它们一起粉碎。假设石头的重量分别为 x 和 y,且 x <= y。那么粉碎的可能结果如下:
如果 x == y,那么两块石头都会被完全粉碎;
如果 x != y,那么重量为 x 的石头将会完全粉碎,而重量为 y 的石头新重量为 y-x。
最后,最多只会剩下一块石头。返回此石头的重量。如果没有石头剩下,就返回 0。
示例:
输入:[2,7,4,1,8,1]
输出:1
解释:
先选出 7 和 8,得到 1,所以数组转换为 [2,4,1,1,1],
再选出 2 和 4,得到 2,所以数组转换为 [2,1,1,1],
接着是 2 和 1,得到 1,所以数组转换为 [1,1,1],
最后选出 1 和 1,得到 0,最终数组转换为 [1],这就是最后剩下那块石头的重量。
提示:
1 <= stones.length <= 30
1 <= stones[i] <= 1000
解题思路:动态规划,每次选出最大的两个值进行操作,将大值置为0小值置为差,继续查找最大的两个元素,直到表中元素遍历完成
func lastStoneWeight(stones []int) int {
//维护变量,数组长度(确定是否便利完全),最大值,第二大值,最大值下标,第二大下标,返回结果,
first, second := 0, 0
firstIndex, secondIndex := -1, -1
result := 0
length := len(stones)
for length > 0 {
//找到最大值,并在数组中将其置为0
firstIndex = findMaxIndex(stones)
first = stones[firstIndex]
stones[firstIndex] = 0
//找到第二大值
secondIndex = findMaxIndex(stones)
second = stones[secondIndex]
result = first - second
//将其置为差值
stones[secondIndex] = result
//数组长度根据最大的两个值是否相同减去相应的长度
if result == 0 {
length = length - 2
} else {
length = length - 1
}
}
return result
}
返回最大值的下标:
//返回最大值的下标
func findMaxIndex(stones []int) int {
max, index := -1, -1
for i := 0; i < len(stones); i++ {
if stones[i] > max {
max = stones[i]
index = i
}
}
return index
}