给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 -1。
示例:
输入: nums = [-1,0,3,5,9,12], target = 9
输出: 4
解释: 9 出现在 nums 中并且下标为 4
-
假设 nums 中的所有元素是不重复的。
-
n 将在 [1, 10000]之间。
-
nums 的每个元素都将在 [-9999, 9999]之间。
前提条件
-
数组为有序数组
-
数组中无重复元素(否则二分查找法返回的元素下标可能不是唯一的)
难点:对区间的定义没有想清楚,区间的定义就是不变量
写二分法,区间的定义一般为两种,左闭右闭即[left, right],或者左闭右开即[left, right)
(1)左闭右闭
-
while (left <= right) 要使用 <= ,因为left == right是有意义的,所以使用 <=
-
if (nums[middle] > target)时 , right 要赋值为 middle - 1,因为当前这个nums[middle]一定不是target,那么接下来要查找的左区间结束下标位置就是 middle - 1
-
if (nums[middle] < target)时, 在右区间,left = middle + 1
class Solution {
public:
int search(vector<int>& nums, int target) {
int left = 0;
int right = nums.size() - 1; // 定义target在左闭右闭的区间里,[left, right]
while (left <= right) { // 当left==right,区间[left, right]依然有效,所以用 <=
int middle = left + ((right - left) / 2);// 防止溢出 等同于(left + right)/2
if (nums[middle] > target) {
right = middle - 1; // target 在左区间,所以[left, middle - 1]
} else if (nums[middle] < target) {
left = middle + 1; // target 在右区间,所以[middle + 1, right]
} else { // nums[middle] == target
return middle; // 数组中找到目标值,直接返回下标
}
}
// 未找到目标值
return -1;
}
};
(2)左闭右开
right是取不到的。
-
while (left < right),这里使用 < ,因为left == right在区间[left, right)是没有意义的
-
if (nums[middle] > target) right 更新为 middle,因为当前nums[middle]不等于target,去左区间继续寻找,而寻找区间是左闭右开区间,所以right更新为middle,即:下一个查询区间不会去比较nums[middle]
-
if (nums[middle] < target) , 寻找右区间,left 更新为middle +1;
class Solution {
public:
int search(vector<int>& nums, int target) {
int left = 0;
int right = nums.size(); // 定义target在左闭右开的区间里,即:[left, right)
while (left < right) { // 因为left == right的时候,在[left, right)是无效的空间,所以使用 <
int middle = left + ((right - left) >> 1);
if (nums[middle] > target) {
right = middle; // target 在左区间,在[left, middle)中
} else if (nums[middle] < target) {
left = middle + 1; // target 在右区间,在[middle + 1, right)中
} else { // nums[middle] == target
return middle; // 数组中找到目标值,直接返回下标
}
}
// 未找到目标值
return -1;
}
};
总结
根据定义区间,确定循环不变量规则。然后在左右区间中,更新left与right。
搜索插入位置
给定一个排序数组(有序不重复)和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
输入: [1,3,5,6], 5
输出: 2
输入: [1,3,5,6], 2
输出: 1
输入: [1,3,5,6], 7
输出: 4
输入: [1,3,5,6], 0
输出: 0
思路
要在数组中插入目标值,无非是这四种情况
-
目标值在数组所有元素之前
-
目标值等于数组中某一个元素
-
目标值插入数组中的位置
-
目标值在数组所有元素之后
1.暴力解法
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
for (int i = 0; i < nums.size(); i++) {
// 分别处理如下三种情况
// 目标值在数组所有元素之前
// 目标值等于数组中某一个元素
// 目标值插入数组中的位置
if (nums[i] >= target) { // 一旦发现大于或者等于target的num[i],那么i就是我们要的结果
return i;
}
}
// 目标值在数组所有元素之后的情况
return nums.size(); // 如果target是最大的,或者 nums为空,则返回nums的长度
}
};
2.二分法
//定义 target 是在一个在左闭右闭的区间里
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int n = nums.size();
int left = 0;
int right = n - 1; // 定义target在左闭右闭的区间里,[left, right]
while (left <= right) { // 当left==right,区间[left, right]依然有效
int middle = left + ((right - left) / 2);// 防止溢出 等同于(left + right)/2
if (nums[middle] > target) {
right = middle - 1; // target 在左区间,所以[left, middle - 1]
} else if (nums[middle] < target) {
left = middle + 1; // target 在右区间,所以[middle + 1, right]
} else { // nums[middle] == target
return middle;
}
}
// 分别处理如下四种情况
// 目标值在数组所有元素之前 [0, -1]
// 目标值等于数组中某一个元素 return middle;
// 目标值插入数组中的位置 [left, right],return right + 1
// 目标值在数组所有元素之后的情况 [left, right], 因为是右闭区间,所以 return right + 1
return right + 1;
}
};
//定义 target 是在一个在左闭右开的区间里,也就是[left, right)
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int n = nums.size();
int left = 0;
int right = n; // 定义target在左闭右开的区间里,[left, right) target
while (left < right) { // 因为left == right的时候,在[left, right)是无效的空间
int middle = left + ((right - left) >> 1);
if (nums[middle] > target) {
right = middle; // target 在左区间,在[left, middle)中
} else if (nums[middle] < target) {
left = middle + 1; // target 在右区间,在 [middle+1, right)中
} else { // nums[middle] == target
return middle; // 数组中找到目标值的情况,直接返回下标
}
}
// 分别处理如下四种情况
// 目标值在数组所有元素之前 [0,0)
// 目标值等于数组中某一个元素 return middle
// 目标值插入数组中的位置 [left, right) ,return right 即可
// 目标值在数组所有元素之后的情况 [left, right),因为是右开区间,所以 return right
return right;
}
};
总结
根据4种情况,利用middle和right的值作为返回值,根据区间定义不同,判断right是否+1;
排序数组中找某元素首尾的位置
待补充ing
移除元素
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
暴力解法
// 时间复杂度:O(n^2)
// 空间复杂度:O(1)
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int size = nums.size();
for (int i = 0; i < size; i++) {
if (nums[i] == val) { // 发现需要移除的元素,就将数组集体向前移动一位
for (int j = i + 1; j < size; j++) {
nums[j - 1] = nums[j];
}
i--; // 因为下标i以后的数值都向前移动了一位,所以i也向前移动一位
size--; // 此时数组的大小-1
}
}
return size;
}
};
时间复杂度:O(n^2)
空间复杂度:O(1)
双指针法
快指针:新数组需要的元素
慢指针:新数组的下标
//不足:需要保留的元素的要重复赋值
int removeElemt(vector<int>& nums, int val) {
int slowIndex = 0;
for (int fastIndex = 0; fastIndex < nums.size(); fastIndex++) {
if (nums[fastIndex] != val) {
cout << nums[fastIndex] << endl;
nums[slowIndex++] = nums[fastIndex];
//slowIndex++;
}
}
nums.resize(slowIndex);//return slowIndex;
return slowIndex;//正好是新数组的大小,此时其它的位置不是数组需要的元素
}
相向双指针方法
基于元素顺序可以改变的题目描述改变了元素相对位置,确保了移动最少元素
int removeElemt2(vector<int>& nums, int val) {
int leftIndex = 0;
int rightIndex = nums.size() - 1;
while (leftIndex <= rightIndex) {
// 找到左边等于val的元素
while (leftIndex <= rightIndex && nums[leftIndex] != val)
{
leftIndex++;
}
// 找到右边不等于val的元素
while (leftIndex <= rightIndex && nums[rightIndex] == val)//等于val时。右边的索引--,就排除了这个元素
{
rightIndex--;
}
//cout << "leftIndex = " << leftIndex << " rightIndex = " << rightIndex << endl;
// 将右边不等于val的元素覆盖左边等于val的元素
if (leftIndex < rightIndex) {
//cout << "变之前的nums[leftIndex] = " << nums[leftIndex] << " 变之前的nums[rightIndex] = " << nums[rightIndex] << endl;
//nums[leftIndex++] = nums[rightIndex--];//先进行赋值操作,再进行++、--
nums[leftIndex] = nums[rightIndex];
leftIndex++;
rightIndex--;
}
}
nums.resize(leftIndex);
return leftIndex;
}
总结day1,这部分内容算是2刷了,但是还是手生,花了3个多小时。