2023.04.12-华子实习笔试-第一题-购物系统的降级策略

原题可以访问codefun2000，原题

直接上代码。。。

import java.util.List;
import java.util.*;
public class Main{
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        String[] str = sc.nextLine().split(" ");
        int[] rr = new int[str.length];
        long sum=0;
        int max=0;
        for (int i = 0; i < str.length; i++) {
            rr[i] = Integer.parseInt(str[i]);
        }
        for (int i = 0; i < rr.length; i++) {
            sum+= rr[i];
            max = Math.max(max,rr[i]);
        }
        int cnt = sc.nextInt();
        if(sum<=cnt) {
            System.out.println(-1);
            return;
        }
        int l=0, r=max;
        int mid=0;
        while(l<r){
            mid=(l+r+1)>>1;
            if(check(rr,cnt,mid)){
                l=mid;
            }
            else{
                r=mid-1;
            }

        }
        System.out.println(l);

    }
    public static boolean check(int[] r, int cnt, int val){
        int n = r.length;
        long ans=0;
        for(int i=0;i<n;i++){
            if(r[i]<val){
                ans+=r[i];
            }
            else{
                ans+=val;
            }
        }
        return ans<=cnt;
    }
}

复杂度控制在O(nlogn）才可以过。

题目意思就是需要找一个界定值，让数组中小于它的维持原值，大于它的等于界定值，并且满足此时的数组之和一定是等于cnt。

求数组被界定值改变之后的和只能通过o(n)去遍历获得，另一步去找出界定值，设为value，可以用二分O(logn）的复杂度解决。因为如果value是最大值解决这个问题的，那么value-1也是符合题意的。

所以先在输入时求出数组的最大值，那么左边界设为0，右边界设为max，开始二分查找，由于小于value符合题意，找右边的临界值，所以在r处我们以mid来更新，而不是mid+1来。

结束循环，即可得出结果。