leetcode 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/last-stone-weight。

解题思路
这里使用排序可以很简单的解决这个问题，这大概也是这道题被归于简单类的原因。（排序可能解法：先使用一个nlogn的排序先排一遍，然后每次取最大的两个元素，如果有剩下就放回，按照插入排序或者冒泡排序的某一步过程使原数组重新有序，重复直到找到答案 即可）

不过同时这也是一个很好的锻炼堆的使用的题。
思路很简单，构造一个最大堆，然后每次分别选择堆顶两个元素（这里取完元素需要整堆），然后模拟题目给出的过程，如果有剩余就要重新加入堆中在，重复直到找到答案。
而且显然这要比直接排序快一些。

这里踩个坑：leetcode里面最好不要用全局变量，否则谁都不知道它会被修改成啥。。。

代码

void swap(int *x, int *y){//交换
    int tmp = *x;
    *x = *y;
    *y = tmp;
}

int abs_(int x, int y){//绝对值
	if(x > y) return x - y;
	else return y - x;
}
void push_down(int v[], int index, int heap_size){//下推
    int max_index = -1;
    int l = index*2, r = index*2+1;
    if(l <= heap_size && v[l] > v[index]) max_index = l;
    else max_index = index;
    if(r <= heap_size && v[r] > v[max_index]) max_index = r;
    if(max_index != index){
        swap(&v[max_index], &v[index]);
        push_down(v, max_index, heap_size);
    }
}

void build_heap(int v[], int heap_size){//建堆
    for(int i = heap_size/2; i >= 1; i--){
        push_down(v, i, heap_size);
    }
}

void pull_up(int v[], int heap_size){//上推
    if(v[heap_size] > v[heap_size/2] && heap_size > 1){
        swap(&v[heap_size], &v[heap_size/2]);
        pull_up(v, heap_size/2);
    }
}

void insert(int v[], int x, int size){//插入
    v[size] = x;
    pull_up(v, size);
}

int get_delete_max(int v[], int size){//选择并删除最大元素
    swap(&v[1], &v[size]);
    int ret = v[size];
    push_down(v, 1, size-1);
    return ret;
}
int lastStoneWeight(int* stones, int stonesSize){
    int s[stonesSize+2];
	for(int i = stonesSize; i > 0; i--){
		s[i] = stones[i-1];
	}//忽略这段，我只是习惯了。。。
	int size = stonesSize;
	build_heap(s, size);
	int ans = s[1];
	while(size > 1){
		int max1 = 0, max2 = 0;
		max1 = get_delete_max(s,size);
        size--;
		max2 = get_delete_max(s,size);
        size--;
		if(max1 != max2){
			int tmp = abs_(max1, max2);
            size++;
			insert(s, tmp, size);
			ans = tmp;
		}
        else{
            if(size > 0) ans = s[1];
            else ans = 0;
        }
	}
	if(size == 1) return ans;
	else return abs_(s[1], s[2]);
}

如有错误，欢迎指出。