Description
现在有n堆果子,第i堆有ai个果子。现在要把这些果子合并成一堆,每次合并的代价是两堆果子的总果子数。求合并所有果子的最小代价。
Input
第一行包含一个整数T(T<=50),表示数据组数。
每组数据第一行包含一个整数n(2<=n<=1000),表示果子的堆数。
第二行包含n个正整数ai(ai<=100),表示每堆果子的果子数。
Output
每组数据仅一行,表示最小合并代价。
Sample Input
2
4
1 2 3 4
5
3 5 2 1 4
Sample Output
19
33
Hint
Source
国防科学技术大学第十八届银河之光文化节ACM程序设计竞赛初赛
1.优先队列
#include<iostream>
#include<cstring>
#include<cstdio>
#include<string>
#include<queue>
using namespace std;
struct node{
int m;
bool operator < (const node &a) const
{
return a.m<m;
}//重载小于运算符,使之以果子少的在前
};
int main()
{
int t;
cin>>t;
int j,n,sum;
priority_queue<node>Q;
node p,q;
while(t--)
{
cin>>n;
for(int i=0;i<n;i++)
{
cin>>p.m;
Q.push(p);
}
sum=0;
for(int i=0;i<n-1;i++)//合并n-1次,每次合并两个最小的果子,弹出队列,把合并的果子重新写进队列
{
p=Q.top();
Q.pop();
q=Q.top();
Q.pop();
sum+=p.m+q.m;
p.m=p.m+q.m;
Q.push(p);
}
while(!Q.empty())
{
Q.pop();
}
cout<<sum<<endl;
}
return 0;
}
2.找最小果子做法
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstdlib>
using namespace std;
int a[1005];
int n;
int my_min(int x)
{
int q,i;
for(q=x,i=q+1;i<n;i++)
{
if(a[i]<a[q])
{
q=i;
}//找出最小的元素
}
swap(a[x],a[q]);//把最小的元素放到x的位置上
}
int main()
{
int t,sum;
cin>>t;
while(t--)
{
cin>>n;
sum=0;
memset(a,0,sizeof(a));
for(int i=0;i<n;i++)
{
cin>>a[i];
}
my_min(0);
my_min(1);
for(int i=1;i<n;i++)
{
a[i]+=a[i-1];
sum+=a[i];
my_min(i);
my_min(i+1);
}
cout<<sum<<endl;
}
}
解题报告:
标准贪心,每次取最小的果子合并后再排序一次,会导致TE。所以我采取的做法是先找出最小的两个,然后把他们加进去省下排序的时间。
更优的解法是优先队列,按果子从小到大排列,把top的两个果子弹出,把合并后的果子压入队列。