31-题目1107：搬水果

http://ac.jobdu.com/problem.php?pid=1107

明明是跟前一道题差不多的，也是求哈夫曼的值，但是就是直接copy代码过来就是不行。。。。可能是对数据的排序有要求。因为如果数据过大，那么不需要每次都从当前位置排到数组的末尾，只需要排两个位置就可以了。但是我这样改之后还是不对。所以我就找了一个建立小根堆的算法，AC了。。。用STL模板做的，希望以后能熟练掌握。

自己的代码：（没有AC，runtime error）

#include<stdio.h>
#include<string.h>
#include<fstream>
#include<algorithm>
#include<iostream>
using namespace std;

int main(){
	int n, i;
	ifstream cin("data.txt");
	while (cin >> n && n != 0)
	{
		int *arr = new int[n];
		for (i = 0; i < n; i++)
			cin >> arr[i];
		sort(arr, arr + n);
		int sum = 0;
		for (i = 1; i < n; i++)
		{
			sum += arr[i - 1] + arr[i];
			arr[i] = arr[i - 1] + arr[i];
			sort(arr + i - 1, arr + i + 1);   //将arr[i-1],arr[n-1]递增排序
		}
		printf_s("%d\n", sum);
	}
	system("pause");
	return 0;
}

小根堆：

//优先队列建立最小堆 AC 40ms
#include <iostream>
#include<fstream>
#include <cstdio>
#include <queue>
#include <functional> // std::greater STL 定义运算函数(代替运算符)
using namespace std;

int main()
{
	int n, a, sum;
	priority_queue<int, vector<int>, greater<int> > q;
	ifstream cin("data.txt");
	while (cin >> n && n != 0)
	{
		sum = 0;
		for (int i = 0; i<n; i++)  
		{
			cin >> a;
			q.push(a);
		}
		while (q.size() >= 2)   //这是小根堆？？？
		{
			int b = q.top();   
			q.pop();
			int c = q.top();
			q.pop();
			q.push(b + c);
			sum += b + c;
		}
		cout << sum << endl;
		q.pop();
	}
	system("pause");
	return 0;
}