C语言---LeetCode 1046. 最后一块石头的重量

1046. 最后一块石头的重量

题目

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

每一回合，从中选出两块 最重的 石头，然后将它们一起粉碎。 假设石头的重量分别为 和 ，且 。 那么粉碎的可能结果如下：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

思路

定义一个函数eliminateMax，获取当前stones数组的最大值max保留并将其对应位置赋值为0。

分别获取两次当前数组的最大值，即调用两次函数eliminateMax，便能获取数组中第一大和第二大值，分别赋值给y,x

然后判断

• 如果x等于0时，数组stones中只有一个石头，返回这个石头的重量。

• x不等于 0，计算y - x的值，并把该值添加到数组stones中，作为新石头的重量

重复上述过程即可得到最终石头的重量。

代码实现

 //寻找数组的最大值，并将其归零
    int eliminateMax(int* nums,int numsSize){
        int max_index = 0;
        int i = 0;
        //获取最大值对应的下标
        for(i = 0; i < numsSize; i++){
            if(nums[max_index] < nums[i]){
                max_index = i;
            }
        }
        //获取最大值
        int val = nums[max_index];
        //将数组中最大值归0
        nums[max_index] = 0;
        return val;
    }
    
    //将val添加到数组中值为0的地方
    void insert(int* nums,int numsSize,int val){
        int i = 0;
        for(i = 0; i < numsSize; i++){
            //当该元素值为0时，将val赋值给nums[i]
            if(0 == nums[i]){
                nums[i] = val;
                return;
            }
        }
    }
    
    int lastStoneWeight(int* stones, int stonesSize){
        while(true){
            int y = eliminateMax(stones,stonesSize);
            int x = eliminateMax(stones,stonesSize);
            //当x == 0时，表示当前数组中只有一个石头，直接返回y
            if(0 == x)
                return y;
            insert(stones,stonesSize,y - x);
        }
    }