牛客网NC75-20.7.24-堆

链接:牛客网NC75
题意
牛客网NC75-20.7.24-堆_第1张图片输入:给定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;
    }
}

你可能感兴趣的:(每日一题)