代码随想录算法训练营第一天 | 704. 二分查找，27. 移除元素

数组处理（二分查找、移除元素）

数组的基础理论知识

题一：二分查找

leetcode 704 二分查找

思路

在二分查找的过程中，需要考虑的是while判断的边界以及left, right指针更新的逻辑。

1. 我们先考虑左闭右闭，即查找的范围在nums[left, right]中。此时当left = right时，仍然有一个元素在查找范围内，因此while的判断边界应该为while left <= right；在left，right指针更新过程中，left应该被更新为mid+1，right应该被更新为mid - 1。代码如下：

class Solution:
    def search(self, nums: List[int], target: int) -> int:
        n = len(nums)
        left, right = 0, n - 1
        mid = (left + right) // 2
        while left <= right:
            if nums[mid] == target:
                return mid
            elif nums[mid] < target:
                left = mid + 1
            else:
                right = mid - 1
            
            mid = (left + right) // 2
        
        return -1

2. 考虑左闭右开情况，在此情况下，right应该被初始化为n，因为待查找的元素范围为[left, right)，相应地，while的判断边界应被修改为left < right，在更新right指针时，应该为right = mid

题2 移除元素

leetcode 27 移除元素

思路

为了原地移除元素，应当维护两个指针：快慢指针。快指针每次均+1，直到快指针遍历完整个数组。当慢指针指向的值为想要移除的元素时，慢指针此次不右移，同时当快指针的值不等于被移除的元素时，将快指针所指向的值赋值给慢指针

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        # 快慢指针
        fast = 0  # 快指针
        slow = 0  # 慢指针
        size = len(nums)
        while fast < size:  # 不加等于是因为，a = size 时，nums[a] 会越界
            # slow 用来收集不等于 val 的值，如果 fast 对应值不等于 val，则把它与 slow 替换
            if nums[fast] != val:
                nums[slow] = nums[fast]
                slow += 1
            fast += 1
        return slow

题3 有序数的平方

leetcode 977 有序数的平方

思路

为了实现O(n)的时间复杂度，考虑采用左右指针进行平方操作：
左指针指向数组开头，右指针指向数组尾部。
建立while循环，循环结束标志为左指针>右指针
判断左右指针指向数的平方大小，将较大的倒叙保存入新数组中，将对应的指针相应地进行移动

class Solution:
    def sortedSquares(self, nums: List[int]) -> List[int]:
        n = len(nums)
        left, right = 0, n - 1
        result = []

        while left <= right:
            sq_left = nums[left] ** 2
            sq_right = nums[right] ** 2

            if sq_left < sq_right:
                result = [sq_right] + result
                right -= 1
            else:
                result = [sq_left] + result
                left += 1
        
        return result