2300. 咒语和药水的成功对数 - 力扣(LeetCode)
给你两个正整数数组
spells和potions,长度分别为n和m,其中spells[i]表示第i个咒语的能量强度,potions[j]表示第j瓶药水的能量强度。同时给你一个整数
success。一个咒语和药水的能量强度 相乘 如果 大于等于success,那么它们视为一对 成功 的组合。请你返回一个长度为
n的整数数组pairs,其中pairs[i]是能跟第i个咒语成功组合的 药水 数目。示例 1:
输入:spells = [5,1,3], potions = [1,2,3,4,5], success = 7 输出:[4,0,3] 解释: - 第 0 个咒语:5 * [1,2,3,4,5] = [5,10,15,20,25] 。总共 4 个成功组合。 - 第 1 个咒语:1 * [1,2,3,4,5] = [1,2,3,4,5] 。总共 0 个成功组合。 - 第 2 个咒语:3 * [1,2,3,4,5] = [3,6,9,12,15] 。总共 3 个成功组合。 所以返回 [4,0,3] 。示例 2:
输入:spells = [3,1,2], potions = [8,5,8], success = 16 输出:[2,0,2] 解释: - 第 0 个咒语:3 * [8,5,8] = [24,15,24] 。总共 2 个成功组合。 - 第 1 个咒语:1 * [8,5,8] = [8,5,8] 。总共 0 个成功组合。 - 第 2 个咒语:2 * [8,5,8] = [16,10,16] 。总共 2 个成功组合。 所以返回 [2,0,2] 。提示:
n == spells.lengthm == potions.length1 <= n, m <= 1051 <= spells[i], potions[i] <= 1051 <= success <= 1010
class Solution {
public int[] successfulPairs(int[] spells, int[] potions, long success) {
//ArrayList<Integer> pairs = new ArrayList<Integer>();
Arrays.sort(potions);
int lenp = potions.length;
int lens = spells.length;
int[] pairs = new int[lens];
for(int i = 0 ; i < lens ; i++) {
int left = 0;
int right = lenp - 1;
while(left < right) {
int mid = (left+right) /2;
long tmp = (long) spells[i] * (long)potions[mid];
if(tmp < success) left=mid+1;
else right = mid;
}
if(right == lenp - 1 && (long) spells[i] * (long)potions[right] < success) {
right+=1;
}
pairs[i] = lenp-right;
}
// //暴力解法
// for(int i = 0; i < lens ; i++) {
// int count = 0 ;
// for(int j = 0; j < lenp; j++) {
// long tmp = (long)spells[i]*(long)potions[j];
// if(tmp >= success) count++;
// }
// pairs.add(count);
// }
//return pairs.stream().mapToInt(Integer::valueOf).toArray();
return pairs;
}
}
每日一题。今天是中等题。
读题,题目的要求很简单,两个数组元素相乘,大于success就算可以,算作1个。那么实际上也是查数的一种。就是target会变化,变化的范围是spells数组的元素。所以每次用spells数组里的元素作为target,对potions数组做二分查找即可。
一开始博主的思路就是暴力解题,两个for循环直接解题,但是力扣直接给我一个超出时间的结果。显然,10的5次方数据量暴力是不行的。那就只能二分,left,right,len三件套,这次使用的是>,也就是right是大于等于target目标数的位置。但有一个地方要注意,就是right==len-1的位置,有可能len-1的位置符合标准,也有可能len-1的位置不是,但没有更大的了。所以这个位置要做一次特殊判断,之后直接使用len-right,就可以快速得出每个位置的答案了。
这道中等题也不难。结果如下:

题目要求计算每个咒语能与多少瓶药水组成能量乘积大于等于给定值的成功组合,通过二分查找优化算法求解。
1061

被折叠的 条评论
为什么被折叠?



