一、题目描述
有一堆石头,每块石头的重量都是正整数。
每一回合,从中选出两块 最重的 石头,然后将它们一起粉碎。假设石头的重量分别为 x 和 y,且 x <= y。那么粉碎的可能结果如下:
如果 x == y,那么两块石头都会被完全粉碎;
如果 x != y,那么重量为 x 的石头将会完全粉碎,而重量为 y 的石头新重量为 y-x。
最后,最多只会剩下一块石头。返回此石头的重量。如果没有石头剩下,就返回 0。
二、解题思路
第一个想法是先将数组排个序,排完序之后数组从小到大的顺序,然后根据题意,取出最重的两块石头,有两种情况考虑:
- 当两块石头一样中的时候,粉碎为0,无需进行插入操作,石头数减少2,将数组长度减2即可
- 当两块石头不一样重时,重量少的石头粉碎为0,重量多的石头剩下一部分,数组总长度减一,重新排序。
当然为了将粉碎为0的石头将其在数组中移除,我们需要进行不为0的判断,数组中最少有两个元素,因为是从后往前取元素的,也就是倒数第二个元素不为0
while(stones[stones.length-2]!=0){
//重置倒数第一块石头
stones[stones.length-1] = stones[stones.length-1]-stones[stones.length-2];
//重置倒数第二块石头,因为while循环的非0判断条件,所以重新排序时0就没有了
stones[stones.length-2] = 0;
Arrays.sort(stones);
}
二、代码演示
class Solution {
public int lastStoneWeight(int[] stones) {
//特判:数组中只有一块石头返回该石头的重量
if(stones.length ==1){
return stones[0];
}
//先将数组排序
Arrays.sort(stones);
while(stones[stones.length-2]!=0){
//重置倒数第一块石头
stones[stones.length-1] = stones[stones.length-1]-stones[stones.length-2];
//重置倒数第二块石头,因为while循环的非0判断条件,所以重新排序时0就没有了
stones[stones.length-2] = 0;
Arrays.sort(stones);
}
return stones[stones.length-1];
}
}