LeetCode—— 1046 最后一块石头的重量

问题描述

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

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

• 如果 x == y，那么两块石头都会被完全粉碎；
• 如果 x != y，那么重量为 x 的石头将会完全粉碎，而重量为 y 的石头新重量为 y-x。

最后，最多只会剩下一块石头。返回此石头的重量。如果没有石头剩下，就返回 0。

提示：

1. 1 <= stones.length <= 30
2. 1 <= stones[i] <= 1000

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/last-stone-weight
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

执行结果

代码描述

思路：

1. 将石头的重量，放在map中，键为重量，值为个数。
2. 由于map天生有序，且按照键 从小到大排序，所以对map倒着扫描。如果这个某个重量是偶数个，则自己消灭自己即可。
3. 如果某个重量是奇数个，则只需要1个石头，与前面的石头进行抵消。
4. 然后把剩余的石头重量，再插进去。
5. 由于差值肯定不会大于当前被消灭的大石头的重量，所以 i 继续挨次往前移动即可。
6. 当扫描完所有的石头之后，从map的头开始，寻找第一个剩下奇数个数的石块，返回其重量。如果没有，就返回0。

补充示例：

输入：1,1,2,4,7,8     输出：1

解释：7,8粉碎，剩1，此时1,1,1,2,4； 2,4粉碎，剩2，此时1,1,1,2；  1,2粉碎，剩1，此时1,1,1；   最终剩1个1.

class Solution {
public:
    int lastStoneWeight(vector<int>& stones) {
        // 将石头插入map中
        map<int, int> wt_cnt;
        for(int i = 0; i < stones.size(); ++i)
        {
            wt_cnt[stones[i]]++;
        }
        // 消灭石头
        for(int i = 1000; i > 0; --i)
        {
            
            if(wt_cnt[i]%2 == 1)
            {
                int j = i-1;
                while(j > 0 && (wt_cnt.count(j) == 0 || wt_cnt[j] == 0))  // 不存在这个重量，或者这个重量的石头为0
                {
                    --j;
                }
                if(j == 0)   break;
                wt_cnt[i-j] += 1;  // 重量差
                wt_cnt[i] = 0;
                wt_cnt[j] -= 1;
            }
        }
        // 寻找剩余最小石头
        for(int i = 1; i < 1001; ++i)
        {
            if(wt_cnt[i]%2 != 0)
                return i;
        }
        return 0;
    }
};