1046. 最后一块石头的重量

1046. 最后一块石头的重量

  • 用STL的堆暴力的解法
class Solution {
public:
    int lastStoneWeight(vector<int>& stones) {

        int len=stones.size();
        if( len<=0 )
        {
            return 0;
        }

        if( 1==len )
        {
            return stones[0];
        }

        while( len>=2 )
        {
            make_heap(stones.begin(),stones.end());
            sort_heap(stones.begin(),stones.end());

            int a=stones[len-1];
            int b=stones[len-2];
            stones.pop_back();
            stones.pop_back();

            if( a!=b )
            {
                stones.push_back( a-b );
                --len;
            }
            else
            {
                len-=2;
            }
        }

        if( 1==len )
        {
            return stones[0];
        }
        else
        {
            return 0;
        }

    }
};
全部评论

相关推荐

Noob1024:一笔传三代,人走笔还在
点赞 评论 收藏
分享
11-15 18:39
已编辑
西安交通大学 Java
全村最靓的仔仔:卧槽,佬啥bg呢,本也是西交么
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务