DAY1 二分查找&移除元素

给定一个 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]之间。

前提条件

  1. 数组为有序数组

  2. 数组中无重复元素(否则二分查找法返回的元素下标可能不是唯一的)

难点对区间的定义没有想清楚,区间的定义就是不变量

写二分法,区间的定义一般为两种,左闭右闭即[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

思路

要在数组中插入目标值,无非是这四种情况

35_搜索插入位置3

  • 目标值在数组所有元素之前

  • 目标值等于数组中某一个元素

  • 目标值插入数组中的位置

  • 目标值在数组所有元素之后

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)
双指针法

快指针:新数组需要的元素

慢指针:新数组的下标

27.移除元素-双指针法

//不足:需要保留的元素的要重复赋值
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个多小时。

  • 17
    点赞
  • 18
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值