代码随想录算法训练营第一天| 704. 二分查找、27. 移除元素、977. 有序数组的平方

数组理论基础

• 数组是存放在连续内存空间上的相同类型数据的集合。

• 数组下标都是从0开始的。

• 数组内存空间的地址是连续的，因此增删需要对整个数组进行操作。

• 数组的元素是不能删的，只能覆盖。

• 在没有指针的数组中，二维数组的空间地址不一定连续，比如java会像图片一样。

 

704. 二分查找

力扣题目链接(opens new window)

给定一个 n 个元素有序的（升序）整型数组 nums 和一个目标值 target ，写一个函数搜索 nums 中的 target，如果目标值存在返回下标，否则返回 -1。

示例 1:

输入: nums = [-1,0,3,5,9,12], target = 9 输出: 4 解释: 9 出现在 nums 中并且下标为 4

1 2 3

示例 2:

输入: nums = [-1,0,3,5,9,12], target = 2 输出: -1 解释: 2 不存在 nums 中因此返回 -1

1 2 3

提示：

• 你可以假设 nums 中的所有元素是不重复的。

• n 将在 [1, 10000]之间。

• nums 的每个元素都将在 [-9999, 9999]之间。

代码

双指针一个在左一个靠右，不断向两个指针中间靠拢。

class Solution:  # class Solution: 这一行是用来定义一个类（class）
    def search(self, nums: list[int], target: int) -> int:  
        # -> int 是函数返回值的类型提示，它说明这个函数将返回一个整数。
        i, j = 0, len(nums) - 1
        while i <= j:
            m = (i + j) // 2
            if nums[m] < target:
                i = m + 1
            elif nums[m] > target:
                j = m - 1
            else:
                return m
        return -1


solution = Solution()
nums = [1, 2, 3, 4, 5]
target = 3
index = solution.search(nums, target)
print(index)  # 输出应该是2

27. 移除元素

力扣题目链接(opens new window)

给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素，并返回移除后数组的新长度。

不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并原地修改输入数组。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

示例 1: 给定 nums = [3,2,2,3], val = 3, 函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。 你不需要考虑数组中超出新长度后面的元素。

示例 2: 给定 nums = [0,1,2,2,3,0,4,2], val = 2, 函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。

你不需要考虑数组中超出新长度后面的元素。

代码

其实看别人写的还有list.pop()或者list.remove()的写法，但我这里还是用快慢指针，其中快指针每次循环走一个，慢指针在快指针所指元素非排除元素才走一格。

class Solution:  # class Solution: 这一行是用来定义一个类（class）
    def removeElement(self, nums: list[int], val: int) -> int:  
    # -> int 是函数返回值的类型提示，它说明这个函数将返回一个整数。
        l, r = 0, 0
        while r < len(nums):
            if nums[r] != val:
                nums[l] = nums[r]
                l += 1  # 所以l直接是新列表的长度
            r += 1
        return l


solution = Solution()
nums = [1, 2, 3, 4, 5]
val = 3
num = solution.removeElement(nums, val)
print(num)  # 输出应该是4

977.有序数组的平方

给你一个按 非递减顺序 排序的整数数组 nums，返回 每个数字的平方 组成的新数组，要求也按 非递减顺序 排序。

示例 1：

• 输入：nums = [-4,-1,0,3,10]

• 输出：[0,1,9,16,100]

• 解释：平方后，数组变为 [16,1,0,9,100]，排序后，数组变为 [0,1,9,16,100]

示例 2：

• 输入：nums = [-7,-3,2,3,11]

• 输出：[4,9,9,49,121]

代码

首先注意列表是按照大小顺序排列的。

其次，因为平方的性质，两头的数平方一定至少有一头比中间的数大。

我首先想着是不新建列表，直接在原列表改动。

右指针大则右指针往左移一格，左指针大则右左指针数互换，右指针往左一格。

但如果原列表中是【-5，-3，-2，-1】这种，会导致出现【1，9，4，25】的情况出现。

因此使用左右指针，每次比较左右指针哪边的数大，放进新列表中。

放进新列表的一边指针移动一格，较小的一方不动。

结束判定条件为左右指针在最后一次判定，重叠后，下一次被判定为交叉结束。

class Solution:

    def sortedSquares(self, nums: list[int]) -> list[int]:
        l, r, i = 0, len(nums) - 1, len(nums) - 1
        res = [float('inf')] * len(nums)  # 需要提前定义列表，存放结果
        while l <= r:
            if nums[l] ** 2 < nums[r] ** 2:  # 左右边界进行对比，找出最大值
                res[i] = nums[r] ** 2
                r -= 1  # 右指针往左移动
            else:
                res[i] = nums[l] ** 2
                l += 1  # 左指针往右移动
            i -= 1  # 存放结果的指针需要往前平移一位
        return res

solution = Solution()
nums = [-5,-3,-2,-1]
res = solution.sortedSquares(nums)
print(res)  # 输出应该是[1,4,9,25]