链接:牛客网NC75
题意:
输入:给定n,c。给定a数组,ai表示第i个数的大小。1<=n<=2*1e5,1<=c<=100,1≤ai≤102
输出:计算的时间
分析:每一次都有两数相加,每次等价于两数之和乘c,而我们可以发现,总的操作次数是固定的,所以为了使时间最小,小的数应该先开始加使得大的数乘c的次数尽可能的少
思路:维护一个小顶堆,通过数组也可以实现,从尾部插入,上浮来维护小顶堆,从头部取得最小值,以尾部元素代替头部元素,从头部元素下沉来维护小顶堆。
看到有人用优先队列priority_queue,这个简单,好像优先队列也是堆实现的吧。
代码:
import java.util.*;
public class Solution {
/**
* 返回一个数字表示输出计算n个数字和的最小花费的时间。
* @param n int整型 表示有n个数。
* @param c int整型 参数c
* @param a int整型一维数组 ai表示第i个数的大小
* @return long长整型
*/
public long solve (int n, int c, int[] a) {
// write code here
int[] b=new int[200005];
int cnt=0;
for(int i=0;i<n;i++){
insert(b,cnt,a[i]);
cnt++;
}
long sum=0;
while(cnt>1){
long tem1=delete(b,cnt);
cnt--;
long tem2=delete(b,cnt);
cnt--;
sum+=(tem1+tem2)*c;
insert(b,cnt,(int)(tem1+tem2));
cnt++;
}
return sum;
}
static void insert(int[] a,int n,int k){
a[n]=k;
if(n==0)return;
int chi=n;
int par=(n-1)/2;
while(par>=0) {
if (a[chi] >= a[par]) break;
int temp = a[chi];
a[chi] = a[par];
a[par] = temp;
chi=par;
par=(chi-1)/2;
}
}
static int delete(int[] a,int n){
int ans=a[0];
a[0]=a[n-1];
int par=0;
int leftchi=1;
int rightchi=2;
while(leftchi<n){
int k=leftchi;
if(rightchi<n&&a[rightchi]<a[leftchi])k++;
if(a[k]>=a[par])break;
int tem=a[k];
a[k]=a[par];
a[par]=tem;
par=k;
leftchi=par*2+1;
rightchi=par*2+2;
}
return ans;
}
}