LCP 18. 早餐组合
解决方案: 排序加二分
假设len(staple) = m,len(drinks) = n
根据数组规模可知,暴力法的时间复杂度为O(mn),会超时。如果数组是有序的就可以用二分法来优化了。于是自然想到先对数组排序。
再多想一步,是不是只需要对其中一个数组排序就可以了呢?如果可以的话要对哪个排序?
只要有一个数组有序,我们遍历另一个无序的数组的每个元素n,在有序的数组中查找第一个比x-n大的数组下标,就是找到了有序数组中有多少个数符合要求。
二分查找的时候要注意边界条件的判断以及极端情况(数组的全部元素都符合要求)。
分析如下:
如果对第一个数组排序,则需要mlogm+nlogm = (m+n)logm
如果对第二个数组排序,则需要nlogn+mlogn = (m+n)logn
所以我们只需要对长度较小的数组排序即可,总的时间复杂度为(m+n)log(min(m,n))
作者:yanghk
链接:https://leetcode-cn.com/problems/2vYnGI/solution/cpython-er-fen-cha-zhao-shuang-100-by-yanghk/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
int helper(vector<int>& v,int n){
if(v.back() <= n) return v.size();
int l = 0,r = v.size()-1;
while(l < r){
int mid = l+(r-l)/2;
if(v[mid] <= n) l = mid+1;
else r = mid;
}
return l;
}
int breakfastNumber(vector<int>& staple, vector<int>& drinks, int x) {
int res = 0;
if(staple.size() <= drinks.size()){
sort(staple.begin(),staple.end());
for(int n: drinks) res = (res+helper(staple,x-n))%1000000007;
}else{
sort(drinks.begin(),drinks.end());
for(int n: staple) res = (res+helper(drinks,x-n))%1000000007;
}return res;
}