石子合并问题 -- 任意版

        有N堆石子，现要将石子有序的合并成一堆，规定如下：每次只能移动任意的2堆石子合并，合并花费为将的一堆石子的数量。设计一个算法，将这N堆石子合并成一堆的总花费最小（或最大）。

此类问题比较简单，就是哈夫曼编码的变形，用贪心算法即可求得最优解。即每次选两堆最少的，合并成新的一堆，直到只剩一堆为止。证明过程可以参考哈夫曼的证明过程。

       代码如下：

#include <iostream>
#include <fstream>
#include <deque>
#include <algorithm>
using namespace std;

struct Node
{
	int key;
	Node* left;
	Node* right;
	Node(){left=0;right=0;}
};

deque<Node*>forest;
ifstream fin("in.txt");
static int min = 0;

bool compare(Node* a,Node* b)
{
	return a->key < b->key;    //求最小组合
	//return a->key > b->key;  //求最大组合
} 

void print(Node *head)
{
	if(head->left)
	{
		cout<<head->key<<"  ";
		print(head->left);
		print(head->right);
		min=min+head->key;
	}
	return;
}

int main()
{
	int n;
	fin>>n;
	Node *p;
	for(int i=0;i<n;i++)
	{
		p=new Node;
		fin>>p->key;
		forest.push_back(p);
	}

/*	//反向迭代器 逆序输出 也可用来求最大值
	deque<Node*>::reverse_iterator rit;
	rit = forest.rbegin();

	for(i=0;i<n;i++)
	{
		cout<<" - "<<rit[i]->key<<endl;
	}
*/
	for(i=0;(i<n) && (forest.size()>1);i++)
	{	
		sort(forest.begin(),forest.end(),compare);
		p = new Node;
		p->key = forest[0]->key  + forest[1]->key;
		p->left = forest[0];
		p->right = forest[1];
		forest.pop_front();
		forest.pop_front();
		forest.push_back(p);
	}
	p = forest.front();
	print(p);
	cout<<endl<<"min:"<<min<<endl;
	
	return 0;
}

输入文件： in.txt 

10
2 4 5 6 4 10 3 6 8 1

输出结果：

49  21  11  28  12  6  3  16  8
min:154
Press any key to continue