Leetcode 15 & 16 (双指针)

时间:2023-01-24 01:12:44

Leetcode 15 & 16 (双指针)

都是比较经典的双指针问题,我们可以从中总结一些双指针的规律


首先这两题如果en做的话就是 \(O(n^{3})\) 的算法,暴力去找。但是我们可以发现这三个值是满足一定约束的,所以考虑使用方法将它降到 \(O(n^2)\)

#Leetcode 15
from ast import List
class Solution:
    def threeSum(self, nums):
        n = len(nums)
        nums.sort()
        if n < 2 or not nums:
            return []
        ans = []
        for i in range(n):
            if nums[i] > 0:
                return ans
            L = i+1
            R = n-1
            if i>0 and nums[i]==nums[i-1]:
                continue
            while (L < R):
                if nums[i]+nums[L]+nums[R] == 0:
                    ans.append([nums[i], nums[L], nums[R]])
                    while (L < R and nums[L+1] == nums[L]):
                        L += 1
                    while (L < R and nums[R-1] == nums[R]):
                        R -= 1
                    L += 1
                    R -= 1
                elif (nums[i]+nums[L]+nums[R] > 0):
                    R -= 1
                else:
                    L += 1
        return ans


if __name__ == "__main__":
    nums = [-1, 0, 1, 2, -1, -4]
    ans = Solution().threeSum(nums=nums)
    print(ans)
#leetcode 16
from ast import List


class Solution:
    def threeSumClosest(self, nums: List, target: int) -> int:
        n = len(nums)
        nums.sort()
        if n < 2 or not nums:
            return []
        ans = 10000000000000000
        for i in range(n):
            L = i+1
            R = n-1
            while (L < R):
                if abs(nums[i]+nums[L]+nums[R]-target)<abs(ans-target):
                    ans = nums[i]+nums[L]+nums[R]
                if nums[i]+nums[L]+nums[R] == target:
                    return target
                elif (nums[i]+nums[L]+nums[R] > target):
                    R -= 1
                else:
                    L += 1
        return ans

# 对于找三个数使其达到所需目标的问题,如果是暴力找的话需要n^3的复杂度
# 但是如果将其排序了,我们就可以使用双指针, 对其进行夹逼
# 因为这三个数是满足一定约束的, 因此可以把复杂度降到 n^2

if __name__ == "__main__":
    nums = [0,0,0]
    target = 1
    ans = Solution().threeSumClosest(nums=nums,target=target)
    print(ans)

未来是什么样,未来会发生什么,谁也不知道。 但是我知道, 起码从今天开始努力, 肯定比从明天开始努力, 要快一天实现梦想。 千里之行,始于足下! ——《那年那兔那些事儿》