合并果子(fruit)（作业1第1题）

Young_20220202

已于 2022-07-09 17:57:44 修改

阅读量118

点赞数

分类专栏：作业1 # 堆文章标签： c++ 算法

于 2022-07-09 17:54:02 首次发布

本文链接：https://blog.csdn.net/Young_20220202/article/details/125696715

版权

作业1 同时被 2 个专栏收录

2 篇文章 0 订阅

订阅专栏

堆

2 篇文章 0 订阅

订阅专栏

链接

题目

Description

在一个果园里，多多已经将所有的果子打了下来，而且按果子的不同种类分成了不同的堆。多多决定把所有的果子合成一堆。

每一次合并，多多可以把两堆果子合并到一起，消耗的体力等于两堆果子的重量之和。可以看出，所有的果子经过n-1次合并之后，就只剩下一堆了。多多在合并果子时总共消耗的体力等于每次合并所耗体力之和。

因为还要花大力气把这些果子搬回家，所以多多在合并果子时要尽可能地节省体力。假定每个果子重量都为1，并且已知果子的种类数和每种果子的数目，你的任务是设计出合并的次序方案，使多多耗费的体力最少，并输出这个最小的体力耗费值。

例如有3种果子，数目依次为1，2，9。可以先将 1、2堆合并，新堆数目为3，耗费体力为3。接着，将新堆与原先的第三堆合并，又得到新的堆，数目为12，耗费体力为 12。所以多多总共耗费体力=3+12=15。可以证明15为最小的体力耗费值。

Format

Input

输入包括两行，第一行是一个整数 n（1\le n\le 30000）n（1≤n≤30000），表示果子的种类数。第二行包含 nn 个整数，用空格分隔，第 ii 个整数 a_i（1\le a_i\le 20000）ai（1≤ai≤20000）是第 ii 种果子的数目。

Output

输出包括一行，这一行只包含一个整数，也就是最小的体力耗费值。输入数据保证这个值小于 2^{31}231。

Samples

输入数据 1

3
1 2 9

输出数据 1

输入数据 2

10
3 5 1 7 6 4 2 5 4 1

输出数据 2

Limitation

1s, 1024KiB for each test case.

题解

一道经典的不能再经典的堆题

可以用STL，可以用手写

自然建议手写

~~因为速度快一点~~因为锻炼一下

投票

函数模板（不包括emplace等）

namespace sheap{
	int n=0,s[100007];
	void s_new_one(){
		n=0;
		memset(s,0,sizeof(s));
	}
	bool s_empty(){
		return (n==0);
	}
	void s_push(int x){
	    int cur=++n,nxt=cur>>1;
	    while(nxt>0){
	    	if(s[nxt]<=x) break;
	    	s[cur]=s[nxt],cur=nxt,nxt>>=1;
		}
		s[cur]=x;
	}
	int s_size(){
		return n;
	}
	int s_front(){
		return s[1];
	}
	bool s_pop(){
		if(n==0) return false;
		if(n==1){
			s[1]=0;
			n--;
			return true;
		}
		s[1]=s[n--];
		int cur=1,nxt=cur<<1;
		int x=s[1];
		while(nxt<=n){
			if(nxt<n&&s[nxt]>s[nxt+1]) nxt++;
			if(s[nxt]>x) break;
			s[cur]=s[nxt],cur=nxt,nxt<<=1;
		}
		s[cur]=x;
		return true;
	}
}

这个便是我们所需的（复制即可）

主函数里面每次取最小的

依次合并即可

全部代码

#include<bits/stdc++.h>
using namespace std;
int n=0,s[100007],l,x;
bool s_empty(){
	return (n==0);
}
void s_push(int x){
	int cur=++n,nxt=cur>>1;
	while(nxt>0){
	    if(s[nxt]<=x) break;
	    s[cur]=s[nxt],cur=nxt,nxt>>=1;
	}
	s[cur]=x;
}
int s_size(){
	return n;
}
int s_front(){
	return s[1];
}
bool s_pop(){
	if(n==0) return false;
	if(n==1){
		s[1]=0;
		n--;
		return true;
	}
	s[1]=s[n--];
	int cur=1,nxt=cur<<1;
	int x=s[1];
	while(nxt<=n){
		if(nxt<n&&s[nxt]>s[nxt+1]) nxt++;
		if(s[nxt]>x) break;
		s[cur]=s[nxt],cur=nxt,nxt<<=1;
	}
	s[cur]=x;
	return true;
}
int main(){
	scanf("%d",&x);
	for(int i=1;i<=x;++i)
	    scanf("%d",&l),s_push(l);
	int ans=0;
	for(int i=n-1;i;--i){
		int t=s_front();
		s_pop();
		if(n) t+=s[1],s_pop();
		ans+=t;
		s_push(t);
	}
	printf("%d",ans);
    return 0;
}