洛谷P1090合併果子AC程式碼(C++)

tlcym發表於2020-11-14

洛谷P1090合併果子AC程式碼(C++)

【題目描述】
在一個果園裡,多多已經將所有的果子打了下來,而且按果子的不同種類分成了不同的堆。多多決定把所有的果子合成一堆。每一次合併,多多可以把兩堆果子合併到一起,消耗的體力等於兩堆果子的重量之和。可以看出,所有的果子經過 n-1次合併之後, 就只剩下一堆了。多多在合併果子時總共消耗的體力等於每次合併所耗體力之和。因為還要花大力氣把這些果子搬回家,所以多多在合併果子時要儘可能地節省體力。假定每個果子重量都為1,並且已知果子的種類 數和每種果子的數目,你的任務是設計出合併的次序方案,使多多耗費的體力最少,並輸出這個最小的體力耗費值。例如有3種果子,數目依次為 1,2,9。可以先將 1 、2 堆合併,新堆數目為 3,耗費體力為 3。接著,將新堆與原先的第三堆合併,又得到新的堆,數目為 12,耗費體力為 12 。所以多多總共耗費體力 3+12=15 。可以證明 15為最小的體力耗費值。

【輸入格式】
共兩行。
第一行是一個整數 n(1≤n≤10000),表示果子的種類數。
第二行包含 n個整數,用空格分隔,第i個整數 ai(1≤ai ≤20000)是第i種果子的數目。

【輸出格式】
一個整數,也就是最小的體力耗費值。輸入資料保證這個值小於 2^31。

【輸入輸出樣例】
輸入:
3
1 2 9
輸出:
15

【資料範圍】
對於30%的資料,保證有n≤1000:
對於50%的資料,保證有n≤5000;
對於全部的資料,保證有n≤10000。

需要注意:

1、體力值是本次合併+上次合併;

2、從小加起消耗最小。

本人使用的是:二叉堆

以下是程式碼:

#include<iostream>
#include<cstdio>
#include<queue>
#include<vector>
using namespace std;
priority_queue<int>heap;
int len=0,ans=0;
void baoli(void)
{
	if(len==1)
	{
		return;
	}
	len--;
	int temp=heap.top();
	heap.pop();
	temp+=heap.top();
	heap.pop();
	ans+=0-temp;
	heap.push(temp);
	baoli();
}
int main(void)
{
	int n;
	scanf("%d",&n);
	len=n;
	for(int i=1;i<=n;i++)
	{
		int x;
		scanf("%d",&x);
		heap.push(-x);
	}
	baoli();
	cout<<ans<<endl;
	return 0;
}

相關文章