一、35. 搜索插入位置
题目描述
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法
代码示例
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int l =0, r = nums.size()-1;
int mid;
while (l <= r) {
mid = l + ( r -l )/2;
if ( nums[mid] == target) {
return mid;
}else if ( nums[mid] < target ) {
l = mid +1;
}else if ( nums[mid] > target ) {
r = mid - 1;
}
}
return r + 1 ;
}
};
解题记录
题目说需要用logN,那肯定是二分查找
并且排序数组,我们不需要自己进行排序,这样我们就不需要计算排序所用的时间了
平平无奇二分查找,插入的位置就是查到的位置的下一位。
二、704. 二分查找
题目描述
给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 -1。
代码示例
class Solution {
public:
int search(vector<int>& nums, int target) {
int l=0,r=nums.size()-1;
int mid;
while(l<r){
mid=l+((r-l)>>1);
if(nums[mid]==target){
return mid;
}else if(nums[mid]>target){
r=mid-1;
}else if(nums[mid]<target){
l=mid+1;
}
}
return nums[l]==target ? l : -1;
}
};
解题记录
这道题我写了得有五六次了,今天头一次进行优化,优化了等于的情况如上代码。
在选定mid
值时,要了解到位运算符>>
的优先级和+
的运算符是一样的,所以要记得加括号,不然就会死循环。
三、剑指 Offer 53 - I. 在排序数组中查找数字 I
题目描述
统计一个数字在排序数组中出现的次数。
代码示例
class Solution {
int binarySearch(vector<int>& nums, int target){
int l=0,r=nums.size()-1;
int mid;
while(l<=r){
mid=l+((r-l)>>1);
if(nums[mid]==target){
return mid;
}else if(nums[mid]>target){
r=mid-1;
}else if(nums[mid]<target){
l=mid+1;
}
}
return -1;
}
public:
int search(vector<int>& nums, int target) {
int idx=binarySearch(nums,target);
if(idx==-1){
return 0;
}
while (idx>0){
if(nums[idx]==nums[idx-1]){
--idx;
}else {
break;
}
}
int cnt=1;
while(idx<nums.size()-1){
if(nums[idx]==nums[idx+1]){
++cnt;
++idx;
}else {
break;
}
}
return cnt;
}
};
解题记录
铁打的二分,流水的offer
实现二分查找函数,
然后用第一个while循环找到第一个为target
的数
然后用第二个while循环计数
四、911. 在线选举
题目描述
给你两个整数数组 persons 和 times 。在选举中,第 i 张票是在时刻为 times[i] 时投给候选人 persons[i] 的。
对于发生在时刻 t 的每个查询,需要找出在 t 时刻在选举中领先的候选人的编号。
在 t 时刻投出的选票也将被计入我们的查询之中。在平局的情况下,最近获得投票的候选人将会获胜。
实现 TopVotedCandidate 类:
TopVotedCandidate(int[] persons, int[] times) 使用 persons 和 times 数组初始化对象。
int q(int t) 根据前面描述的规则,返回在时刻 t 在选举中领先的候选人的编号。
代码示例
class TopVotedCandidate {
vector<int> tops;
vector<int> m_times;
public:
TopVotedCandidate(vector<int>& persons, vector<int>& times) {
unordered_map<int, int> voteCounts;
voteCounts[-1] = -1;
int top = -1;
for (auto & p : persons) {
++voteCounts[p];
if (voteCounts[p] >= voteCounts[top]) {
top = p;
}
tops.emplace_back(top);
}
this->m_times = times;
}
int q(int target){
int idx=upper_bound(m_times.begin(),m_times.end(),target)-m_times.begin()-1;
return tops[idx];
}
};
解题记录
头一次写这种要预处理的题,不是很熟练,二分查找看题解用的api,不可取hhh,
先预处理求和,求每个时间段的最大值
然后用二分查找按时间查找最大值