LeetCode 2300. 咒语和药水的成功对数(排序+二分)

文章目录

LeetCode、2300. 咒语和药水的成功对数【中等,排序+二分】

来源:LeetCode专题《LeetCode 75》

题目及类型

题目链接:LeetCode、2300. 咒语和药水的成功对数

类型:基础算法/二分


思路及代码

思路:

首先对药水能量强度数组进行排序,接着我们去遍历所有的咒语,接着我们对药水能量数组进行二分,通过使用咒语与药水能量乘积来进行二分寻找边界值,来确定成功对数。

image-20240119212337446

代码:

复杂度分析:时间复杂度O(n.logn);空间复杂度O(1)

class Solution {

    //排序+二分
    //找到最右边不成功的组合的最后一个位置
    public int[] successfulPairs(int[] spells, int[] potions, long success) {
        int n = spells.length, m = potions.length;
        //结果数组
        int[] res = new int[n];
        //对potions升序
        Arrays.sort(potions);
        //遍历所有的药水
        for (int i = 0; i < n; i ++) {
            int ans = 0;
            //二分
            int l = 0, r = m - 1;
            while (l < r) {
                //若是拆分为[l, mid - 1]、[mid, r],那么一旦有r = mid - 1,需要+1为l + r + 1
                int mid = (l + r + 1) >> 1;
                // System.out.printf("mid=%d\n", mid);
                if (check(1L * spells[i] * potions[mid], success)) {
                    r = mid - 1;
                }else {
                    l = mid;
                }
            }
            //找到目标值
            // System.out.printf("%d %d\n", l, r);
            //全部成功(根据r的情况值来判定)
            if (r < 0 || (r == 0 && 1L * spells[i] * potions[0] >= success)) {
                ans = m;
            }else if (l >= m || (l == m - 1 && 1L * spells[i] * potions[m - 1] < success)) { //全部不匹配(根据l的情况值来判定)
                ans = 0;
            }else {
                //若是找到区间范围的
                ans = m - l - 1;
            }
            //将结果添加到结果集
            res[i] = ans;
        }
        return res;
    }

    //检测
    public boolean check(long cur, long success) {
        if (cur >= success) return true;
        return false;
    }

}

image-20240119212201819


整理者:长路 整理时间:2024.1.19

评论区请在客户端页面查看