[wikioi]合并果子

http://wikioi.com/problem/1063/

这题是贪心+堆。主要想练习一下堆的写法。算法导论里的方法名是heapify()等，但大家经常用更直观的down(), up()方法（向上，下调整），根据这两个方法，可以有build，insert，getmin方法。

向下调整的代码稍微需要判断一下左右子树是否越界，其他都很简单。

#include <iostream>

using namespace std;

#define LEN 10005

int num[LEN];

int size;



void swap(int a, int b) {

    int tmp = num[a];

    num[a] = num[b];

    num[b] = tmp;

}



void down(int i) {

    if (i > size) return;

    int left = i * 2;

    int right = i * 2 + 1;

    int minIdx = left;

    if (right <= size && num[left] > num[right]) {

        minIdx = right;

    }

    if (left > size || num[i] <= num[minIdx])

        return;

    else {

        swap(i, minIdx);

        down(minIdx);

    }

}



void up(int i) {

    if (i == 1) return;

    if (num[i / 2] > num[i]) {

        swap(i / 2, i);

        up(i / 2);

    }

}



void build() {

    for (int i = size / 2; i >= 1; i--) {

        down(i);

    }

}



int getmin() {

    swap(1, size);

    size--;

    down(1);

    return num[size + 1];

}



void insert(int x) {

    size++;

    num[size] = x;

    up(size);

}



int main() {

    int n;

    cin >> n;

    for (int i = 1; i <= n; i++) {

        cin >> num[i];

    }

    size = n;

    build();

    int sum = 0;

    while (size > 1) {

        int a = getmin();

        int b = getmin();

        sum += a + b;

        insert(a + b);

    }

    cout << sum << endl;

}