哈夫曼递归问题--最小代价切割金条

题目出自左神算法初级班第四期第八个视频

#pragma once
//问题:切金条问题
//关键词:贪心、哈夫曼编码
//描述:一块金条切成两半,需要花费和长度数值一样的铜板。求最省铜板的切金条方法
//     例如,给定数组[10,20,30],代表要把长度为10+20+30=60的金条分成10,20,30三个部分
//     如果先把长度为60的金条分成10和50,花费60,
//     再把长度为50的金条分成20和30,花费30,总共花费110铜板
//     但如果先把长度为60的金条分成30和30,花费60,
//     再把长度为30的金条分成10和20,花费30,总共花费90铜板
//要求:数组一个数组,返回分割的最小代价
//分析:这道题目属于经典的哈夫曼编码问题,建立小根堆,每次从小根堆的堆顶取走两个元素
//      再将这两个元素的和放进小根堆中,直到小根堆只剩下一个元素,即为分割的最小代价
//思考:当一个问题是求代价时,它总共的代价是由子代价的某种计算形成的
//      这种问题都可以考虑采用哈夫曼编码的贪心策略去解决

#include 
#include 
#include 
#include 

using namespace std;

class Huffman
{
public:
	Huffman() {};
	~Huffman() {};
	priority_queue , greater > lessest;

	int minCost(vector arr)
	{
		int cost = 0;
		if (arr.size()==0)
		{
			return 0;
		}
		else if(arr.size()==1)
		{
			return arr[0];
		}
		else if (arr.size() == 2)
		{
			return arr[0] + arr[1];
		}
		for (int i = 0; i < arr.size(); i++)
		{
			lessest.push(arr[i]);
		}
		while (lessest.size()>1)
		{
			int less1 = lessest.top();
			lessest.pop();
			int less2 = lessest.top();
			lessest.pop();
			lessest.push(less1 + less2);
			cost += (less1 + less2);
		}
		return cost;
	}
};


vector arrToVecarr(int *arr, int size)
{
	vector result;
	for (int i = 0; i < size; i++)
	{
		result.push_back(arr[i]);
	}
	return result;
}

#include"cutGoldBar_greedy_huffman.h"
#include 

using namespace std;

int main()
{
	//test cutGoldBar_greedy_huffman.h
	int arr[3] = { 10,20,30 };
	int *arrP = arr;
	vector vecarr = arrToVecarr(arrP, 3);
	Huffman huffman;
	cout << huffman.minCost(vecarr) << endl;
	
	system("pause");
	return 0;
	
}

你可能感兴趣的:(程序员代码面试指南)