洛谷1334 瑞瑞的木板

题目描述

瑞瑞想要亲自修复在他的一个小牧场周围的围栏。他测量栅栏并发现他需要N（1≤N≤20,000）根木板，每根的长度为整数Li（1≤Li≤50,000）。于是，他神奇地买了一根足够长的木板，长度为所需的N根木板的长度的总和，他决定将这根木板切成所需的N根木板。（瑞瑞在切割木板时不会产生木屑，不需考虑切割时损耗的长度)瑞瑞切割木板时使用的是一种特殊的方式，这种方式在将一根长度为x的模板切为两根时，需要消耗x个单位的能量。瑞瑞拥有无尽的能量，但现在提倡节约能量，所以作为榜样，他决定尽可能节约能量。显然，总共需要切割N-1次，问题是，每次应该怎么切呢？请编程计算最少需要消耗的能量总和。

输入格式：

第一行: 整数N，表示所需木板的数量

第2到N+1行: 每行为一个整数，表示一块木板的长度

输出格式：

一个整数，表示最少需要消耗的能量总和

输入样例#1： 

3
8
5
8

输出样例#1： 

34

典型贪心。将最小的两块板子合并，其所消耗能量最少。用一个优先队列来维护板子长度，弹出最小的两块板子合并再放入队列。注意数据过大需要用LL。

#include<iostream>
#include<queue>
#include<functional>
#define ll long long
using namespace std;
int main()
{
	ll n, ans = 0;
	cin >> n;
	priority_queue<ll,vector<ll>,greater<ll> > q;
	for (int i = 0; i < n; i++)
	{
		ll tmp;
		cin >> tmp;
		q.push(tmp);
	}
	for (int i = 0; i < n - 1; i++)
	{
		ll tmp1 = q.top();
		q.pop();
		ll tmp2 = q.top();
		q.pop();
		ll tmp3 = tmp1 + tmp2;
		ans += tmp3;
		q.push(tmp3);
	}
	cout << ans << endl;
	return 0;
}