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.length
m == potions.length
1 <= n, m <= 105
1 <= spells[i], potions[i] <= 105
1 <= 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,就可以快速得出每个位置的答案了。
这道中等题也不难。结果如下: