luogu1090 合并果子(NOIP2004提高组第2题)
时空限制 1000ms/128MB
题目描述
在一个果园里,多多已经将所有的果子打了下来,而且按果子的不同种类分成了不同的堆。多多决定把所有的果子合成一堆。
每一次合并,多多可以把两堆果子合并到一起,消耗的体力等于两堆果子的重量之和。可以看出,所有的果子经过 n−1次合并之后, 就只剩下一堆了。多多在合并果子时总共消耗的体力等于每次合并所耗体力之和。
因为还要花大力气把这些果子搬回家,所以多多在合并果子时要尽可能地节省体力。假定每个果子重量都为 1 ,并且已知果子的种类 数和每种果子的数目,你的任务是设计出合并的次序方案,使多多耗费的体力最少,并输出这个最小的体力耗费值。
例如有 3 种果子,数目依次为 1 , 2 , 9 。可以先将 1 、 2 堆合并,新堆数目为3 ,耗费体力为3 。接着,将新堆与原先的第三堆合并,又得到新的堆,数目为 12 ,耗费体力为 12 。所以多多总共耗费体力 =3+12=15。可以证明 15 为最小的体力耗费值。
输入输出格式
输入格式:
输入文件 fruit.in 包括两行,第一行是一个整数 n(1≤n≤10000),表示果子的种类数。第二行包含 n个整数,用空格分隔,第 i 个整数 ai(1≤ai≤20000)是第 i 种果子的数目。
输出格式:
输出文件 fruit.out 包括一行,这一行只包含一个整数,也就是最小的体力耗费值。输入数据保证这个值小于 2^31 。
输入输出样例
输入样例#1:
3
1 2 9
输出样例#1:
15
说明
对于30%的数据,保证有n<=1000:
对于50%的数据,保证有n<=5000;
对于全部的数据,保证有n<=10000。
代码
法一:插入排序
#include<iostream>
#include<algorithm>
using namespace std;
const int N = 10005;
int n,a[N],ans=0;
int main(){
ios::sync_with_stdio(false);
cin>>n;
for (int i=1; i<=n; i++) cin>>a[i];
sort(a+1,a+n+1);
for (int i=2,x,j; i<=n; i++){
x = a[i-1]+a[i];
ans += x;
for (j=i+1; j<=n; j++)
if (a[j]<x) a[j-1]=a[j];
else break;
a[j-1] = x;
}
cout<<ans<<endl;
return 0;
}
法二:STL优先队列
#include<iostream>
#include<queue>
using namespace std;
priority_queue<int,vector<int>,greater<int> > pq; //小顶堆
int main(){
ios::sync_with_stdio(false);
int n,ans=0;
cin>>n;
for (int i=1,x; i<=n; i++){
cin>>x;
pq.push(x);
}
for (int i=1,x,y; i<n; i++){
x=pq.top(); pq.pop();
y=pq.top(); pq.pop();
ans += x+y;
pq.push(x+y);
}
cout<<ans<<endl;
return 0;
}
法三:手动堆排
#include<iostream>
#include<algorithm>
using namespace std;
const int N = 10005;
int n,a[N],ans=0;
void adjust(int i,int m){ //调整为小顶堆
for (int j=2*i; j<=m; ){
if (j<m && a[j+1]<a[j]) j++;
if (a[j]<a[i]){
swap(a[i],a[j]);
i = j;
j = 2*i;
}
else break;
}
}
void heapsort(int n){
for (int i=n/2; i>=1; i--) adjust(i,n);
for (int i=n; i>1; i--){
swap(a[1],a[i]);
adjust(1,i-1);
a[1] += a[i]; //次小+最小
ans += a[1];
adjust(1,i-1);
}
}
int main(){
ios::sync_with_stdio(false);
cin>>n;
for (int i=1; i<=n; i++) cin>>a[i];
heapsort(n);
cout<<ans<<endl;
return 0;
}
法四:双队列
#include<iostream>
#include<algorithm>
using namespace std;
const int N = 10005, oo = 0x7fffffff;
int a[N],b[N];
int main(){
ios::sync_with_stdio(false);
int n,ans;
cin>>n;
fill(a,a+N,oo);
fill(b,b+N,oo);
for (int i=1; i<=n; i++) cin>>a[i];
sort(a+1,a+n+1);
ans = b[1] = a[1]+a[2];
int i=3,j=1,k=1;
for (int x=1; x<n-1; x++){
if (i<n && a[i+1]<b[j]) ans+=(b[++k]=a[i]+a[i+1]),i+=2;
else if (j<k && b[j+1]<a[i]) ans+=(b[++k]=b[j]+b[j+1]),j+=2;
else ans+=(b[++k]=(a[i]+b[j])),i++,j++;
}
cout<<ans<<endl;
return 0;
}