数组操作大多利用指针来控制范围,将一些经典解法总结于此。
给定一个包含红色、白色和蓝色,一共 n 个元素的数组,对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
注意:
不能使用代码库中的排序函数来解决这道题。示例:
输入: [2,0,2,1,1,0]输出: [0,0,1,1,2,2]
如上图我可以设置令[0,i]区间存放0,[j,len(nums)-1]存放2,k为当前元素。如果nums[k]==1,则i继续加1就好,如果==0,则与i+1位置的元素交换,i和k依次加1。如果==2,则和j-1的位置交换,j-1即可。注意我们的区间是左闭又闭区间,所以i初始化为-1,不能为0!同理j初始不能为len(nums)-1,应该为len(nums).
1 class Solution: 2 def sortColors(self, nums: List[int]) -> None: 3 """ 4 Do not return anything, modify nums in-place instead. 5 """ 6 i,j = -1,len(nums) # 注意初始化 7 k=0 8 while(k
2. leetcode 80.
给定一个排序数组,你需要在删除重复出现的元素,使得每个元素最多出现两次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在修改输入数组并在使用 O(1) 额外空间的条件下完成。
示例 1:
给定 nums = [1,1,1,2,2,3],函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1,1, 2, 2, 3 。你不需要考虑数组中超出新长度后面的元素。
示例 2:
给定 nums = [0,0,1,1,1,1,2,3,3],函数应返回新长度 length = 7, 并且原数组的前五个元素被修改为 0,0, 1, 1, 2, 3, 3 。你不需要考虑数组中超出新长度后面的元素。
i为当前符合条件的下标,k为当前考虑的元素。那么何时才能够加入?应该是当k比其之前的2个位置的元素大时才可以加入。即nums[k]>nums[i-2]时才可以加入。
1 class Solution:2 def removeDuplicates(self, nums: List[int]) -> int:3 i = 04 for n in nums:5 if i<2 or n>nums[i-2]:6 nums[i] = n7 i+=18 return i9
3. leetcode问题
双指针法。两个指针分别向中间移动。满足不同的条件分别移动左指针和右指针。