二分查找
链接: 704.二分查找
思考过程:
思路是二分法,很经典的题目,需要注意的是区间要保持一致
二分法使用的前提条件是数据有序和数据不重复出现
解题代码
int search(vector<int> &data, int target)
{
int left = 0, right = data.size() - 1;
while (left <= right) {
int mid = (right - left) / 2 + left;
if (data[mid] == target) {
return mid;
}
else if (data[mid] > target) {
right = mid - 1;
}
else if (data[mid] < target) {
left = mid + 1;
}
}
return -1;
}
要点总结:注意区分区间的左右开闭处理
移除数组内target元素
链接: 27. 移除元素
我的思路是从数组末端往前遍历,找到target元素,然后将其后的元素都前移一位
时间复杂度:O(n^2)
解题代码如下:
int removeElement(vector<int> &data, int target)
{
int moveSize = data.size();
for (int i = data.size() - 1; i >= 0; i--) {
// for (auto i = data.size() - 1; i >= 0; i--) {
if (data[i] == target) {
// 后面元素前移
int j = i;
while (j < moveSize - 1) {
data[j] = data[j + 1];
j++;
}
moveSize--;
}
}
return moveSize;
}
扩展:
在不同编译器下
for (auto i = data.size() - 1; i >= 0; i–)
中的 变量i 是auto类型,会被不同编译器推导为不同的类型,因为vector中size函数的返回值是一个无符号类型,当size是0的时候,对其进行–,会造成数字的越界溢出。
经过测试,mingw下是未定义类型,msvc下是unsigned __int64类型,
经过优化,可以做到O(n)的时间复杂度
使用双指针来解决这个问题,思路是left增加找到等于val的值,right减少找到第一个不为val的值,left和right处的元素两者交换,当left和right指针相遇,就代表数组所有的val都已经移动到后面了。
优化后的代码:
int removeElement2(vector<int> &nums, int val)
{
int right = nums.size() - 1;
int left = 0;
while (left <= right) {
// find val
while (left <= right && nums[left] != val) left++;
// find not val
while (left <= right && nums[right] == val) right--;
// swap
if (left < right) {
std::swap(nums[left++], nums[right--]);
}
}
return left;
}
注意循环的 left <= right 条件判断要在前面,边界处理的时候要尤其小心