leetcode刷题【四】：数组篇

题目：三数之和

难度：中等

给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有和为 0 且不重复的三元组。
注意：答案中不可以包含重复的三元组。

示例：
输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]

代码：

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        n = len(nums) # 获取数组长度
        nums.sort() # 排序降低时间复杂度
        ans = list() # 创建新的list，存储结果
        
        for i in range(n):
            # 排序后相邻两数如果相等，则跳出当前循环，相同的数字仅需计算一次
            if i > 0 and nums[i] == nums[i-1]:
                continue
            j = i + 1 # 记录i的下一个位置
            k = n - 1 # 最后一个元素的位置
            while j < k :
                if nums[j] + nums[k] == - nums[i]:
                    # 结果加入列表
                    ans.append([nums[i],nums[j],nums[k]])
                    while j < k and nums[j] == nums[j+1]: j += 1
                    while j < k and nums[k] == nums[k-1]: k -= 1
                    # 没有相等则j+1,k-1缩小范围
                    j += 1
                    k -= 1
                elif nums[j] + nums[k] < - nums[i]:
                    j += 1
                else:
                    k -= 1
        return ans

双指针的用法，比较容易漏掉的是如果 nums[j] + nums[k] < - nums[i]: 要让j继续右移。
时间复杂度：$O(n^2)$
空间复杂度：$O(1)$

附上Leetcode官方的解析：

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        n = len(nums)
        nums.sort()
        ans = list()
        
        # 枚举 a
        for first in range(n):
            # 需要和上一次枚举的数不相同
            if first > 0 and nums[first] == nums[first - 1]:
                continue
            # c 对应的指针初始指向数组的最右端
            third = n - 1
            target = -nums[first]
            # 枚举 b
            for second in range(first + 1, n):
                # 需要和上一次枚举的数不相同
                if second > first + 1 and nums[second] == nums[second - 1]:
                    continue
                # 需要保证 b 的指针在 c 的指针的左侧
                while second < third and nums[second] + nums[third] > target:
                    third -= 1
                # 如果指针重合，随着 b 后续的增加
                # 就不会有满足 a+b+c=0 并且 b<c 的 c 了，可以退出循环
                if second == third:
                    break
                if nums[second] + nums[third] == target:
                    ans.append([nums[first], nums[second], nums[third]])
        
        return ans

作者：LeetCode-Solution
链接：https://leetcode-cn.com/problems/3sum/solution/san-shu-zhi-he-by-leetcode-solution/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

题目：移除元素

难度：简单

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

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

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

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        while val in nums:
            nums.remove(val)
        return len(nums)

思考：
这道题使用python做，其实有点偷懒的意思。可以直接使用list的函数remove某个列表中的元素，所以只需要加上条件判断即可。

附：list的函数
list.append(obj)
在列表末尾添加新的对象
list.count(obj)
统计某个元素在列表中出现的次数
list.extend(seq)
在列表末尾一次性追加另一个序列中的多个值（用新列表扩展原来的列表）
list.index(obj)
从列表中找出某个值第一个匹配项的索引位置
list.insert(index, obj)
将对象插入列表
list.pop([index=-1])
移除列表中的一个元素（默认最后一个元素），并且返回该元素的值
list.remove(obj)
移除列表中某个值的第一个匹配项
list.reverse()
反向列表中元素
list.sort(cmp=None, key=None, reverse=False)
对原列表进行排序

其它解法（C++）：

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        # 快慢指针
        if not nums:
            return 0
        # 排序
        nums.sort()
        slow, fast = 0, 0
        # fast越界即停止
        while fast < len(nums):
            if nums[fast] != val:
                # 将不重复的值按顺序赋值
                nums[slow] = nums[fast]
                slow += 1
            fast += 1
        return slow

题目：删除有序数组中的重复项

难度：简单

给你一个有序数组 nums ，请你 原地 删除重复出现的元素，使每个元素 只出现一次 ，返回删除后数组的新长度。

不要使用额外的数组空间，你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。

class Solution:
    def removeDuplicates(self, nums):
        i = 0
        l = 1
        while l < len(nums):
            if nums[i] == nums[l]:
                nums.pop(i)
            else:
                i += 1
                l += 1
        return len(nums)

参考：

删除有序数组中的重复项
【leetcode】80.删除排序数组中的重复项（python实现）