最后一块石头的重量

最后一块石头的重量

• 贪心算法
• 递归

题目描述

有一堆石头，每块石头的重量都是正整数。

每一回合，从中选出两块 最重的 石头，然后将它们一起粉碎。假设石头的重量分别为 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

解题思路

若数组长度为 n , n>2 时，排序后直接操作原数组
【1，2，3，4】粉碎一次变成【1，2，0，1】递归后先排序为
【0，1，1，2】粉碎第二次为【0，1，0，2】递归后先排序为
【0，0，1，2】此时触发递归结束条件，直接返回2-1=1

代码实现

public class 最后一块石头的重量 {
    public static void main(String[] args) {
        int[] stone = {2,7,4,1,8,1};
        int stoneWeight = lastStoneWeight(stone);
        System.out.println("stoneWeight = " + stoneWeight);
    }
    public static int lastStoneWeight(int[] stones) {
        //1.根据最后的索引值来操作最后返回石头的重量
        int stoneIndex = stones.length;
        //2.数组只有一个元素,直接返回
        if (stoneIndex == 1) return stones[0];
        //3.数组有两个元素,直接返回 最大值减去最小值的结果的绝对值Math.abs
        if (stoneIndex == 2) return Math.abs(stones[0]-stones[1]);
        //4.选出两块石头,石头的重量分别为 x 和 y，且 x <= y,那么可以先排序
        Arrays.sort(stones);//默认升序排序
        //5.递归结束的条件,当排序后,倒数第三个元素为0,说明只剩下两个石头,返回最后一个-倒数第二个
        if (stones[stoneIndex-3] == 0) return stones[stoneIndex-1]-stones[stoneIndex-2];
        //6.直接操作原数组,最后一块等于减去前一块,前一块抵消重量后归0
        stones[stoneIndex-1] = stones[stoneIndex-1]-stones[stoneIndex-2];
        stones[stoneIndex-2] = 0;
        //7.递归,重复执行以上动作
        return lastStoneWeight(stones);
    }
}