数马笔试算法10.15

小红拿到了一个长度为n的数组 a,每次操作小红可以选择数组中的任意一个数减去 x,小红一共能进行 k 次。小红想在 k 次操作之后,数组的最大值尽可能小。请你返回这个最大值。

n为1~10^5
a,k,x为1~10^9

示例输入:
5  3  5
1  4  3  11  2

输入为n,k,x 以及数组a的值

示例输出:
3

想知道这道题有没有比堆排更好的做法,我堆排一直tle。

import java.util.PriorityQueue;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();
        long k = scanner.nextLong();
        long x = scanner.nextLong();
        long[] a = new long[n];
        PriorityQueue<Long> pq = new PriorityQueue<>((o1, o2) -> o2.compareTo(o1));
        for(int i =0;i<n;i++){
            a[i] = scanner.nextLong();
            pq.add(a[i]);
        }
        while(k>0){
            Long top = pq.poll();
            Long nextTop = pq.peek();
            Long dif = top-nextTop;
            Long num = dif/x;
            num = Math.min(k-1,num);
            k-=(num+1);
            top-=(num+1)*x;
            pq.add(top);
        }
        System.out.println(pq.peek());

    }
}
全部评论
佬收到ai面了吗,我现在还没消息
点赞 回复 分享
发布于 2024-10-18 10:11 广东
二分答案
点赞 回复 分享
发布于 2024-10-16 00:01 山东
同优先队列,怒得1分
点赞 回复 分享
发布于 2024-10-15 20:54 广东

相关推荐

白火同学:1、简历可以浓缩成一页,简历简历先要“简”方便HR快速过滤出有效信息,再要“历”用有效信息突出个人的含金量。 2、教育背景少了入学时间~毕业时间,HR判断不出你是否为应届生。 3、如果你的平台账号效果还不错,可以把账号超链接或者用户名贴到对应位置,一是方便HR知道你是具体做了什么内容的运营,看到账号一目了然,二是口说无凭,账号为证,这更有说服力。
面试被问期望薪资时该如何...
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务