update: some array topics

Author: junbinding

arr.container-with-most-water.py

@@ -0,0 +1,25 @@
+class Solution:
+    """
+    盛最多水的容器
+    https://leetcode-cn.com/problems/container-with-most-water/
+    给你 n 个非负整数 a1，a2，...，an，每个数代表坐标中的一个点 (i, ai) 。在坐标内画 n 条垂直线，垂直线 i 的两个端点分别为 (i, ai) 和 (i, 0)。
+    找出其中的两条线，使得它们与 x 轴共同构成的容器可以容纳最多的水。
+    说明：你不能倾斜容器，且 n 的值至少为 2。
+    """
+    def maxArea(self, height):
+        maxArea = 0
+        i = 0
+        j = len(height) - 1
+        while(i < j):
+            minHeight = min(height[i], height[j])
+            maxArea = max((j - i) * minHeight, maxArea)
+            if height[i] < height[j]:
+                i += 1
+            else:
+                j -= 1
+
+        return maxArea
+
+so = Solution()
+print(so.maxArea([1,8,6,2,5,4,8,3,7]))
+

arr.move-zeroes.py

@@ -0,0 +1,22 @@
+from typing import List
+
+class Solution:
+    """
+    283. 移动零
+    https://leetcode-cn.com/problems/move-zeroes/
+    给定一个数组 nums，编写一个函数将所有 0 移动到数组的末尾，同时保持非零元素的相对顺序。
+    Do not return anything, modify nums in-place instead.
+    """
+    def moveZeroes(self, nums: List[int]) -> None:
+        idx = 0
+        for i in range(len(nums)):
+            if nums[i] != 0:
+                nums[idx], nums[i] = nums[i], nums[idx]
+                idx += 1
+
+
+so = Solution()
+nums = [0, 1, 0, 3, 12]
+so.moveZeroes(nums)
+print(nums)
+

arr.three-sum.py

@@ -0,0 +1,75 @@
+from typing import List
+
+class Solution:
+    """
+    15. 三数之和
+    https://leetcode-cn.com/problems/3sum/
+    给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有满足条件且不重复的三元组。
+    注意：答案中不可以包含重复的三元组。
+    """
+    # 双指针
+    def threeSum(self, nums: List[int]) -> List[List[int]]:
+        # 若数组长度小于3，则返回空数组
+        res = []
+        n = len(nums)
+        if n < 3:
+            return res
+
+        # 对数组进行排序
+        nums.sort()
+        for i in range(n-2):
+            # 若 i 大于 0 则终止循环，返回结果
+            if nums[i] > 0:
+                return res
+
+
+            # 对于重复元素，直接跳过
+            if i > 0 and nums[i] == nums[i-1]:
+                continue
+
+            # 声明左指针 L=i+1, 右指针 R=n-1,
+            l = i + 1
+            r = n - 1
+
+            while(l < r):
+                num = nums[i] + nums[l] + nums[r]
+                # 若 nums[i] + nums[L] + nums[R] = 0
+                if num == 0:
+                    res.append([nums[i], nums[l], nums[r]])
+                    l += 1
+                    while nums[l] == nums[l-1] and l < r:
+                        l += 1
+                    r -= 1
+                    while nums[r] == nums[r+1] and l < r:
+                        r -= 1
+                # 若和小于0，则说明 nums[L] 小了，L 向右移
+                elif num < 0:
+                    l += 1
+                # 若和大于0，则说明 nums[R] 打了，R 向左移
+                else:
+                    r -= 1
+        return res
+
+    # 暴力破解
+    def threeSumByForce(self, nums: List[int]) -> List[List[int]]:
+        res = []
+        nums.sort()
+        for i in range(len(nums) - 2):
+            if i > 0 and nums[i] == nums[i - 1]:
+                continue
+            for j in range(i + 1, len(nums) - 1):
+                if j > i + 1 and nums[j] == nums[j - 1]:
+                    continue
+                for m in range(j + 1, len(nums)):
+                    if m > j + 1 and nums[m] == nums[m - 1]:
+                        continue
+                    if nums[i] + nums[j] + nums[m] == 0:
+                        res.append([nums[i], nums[j], nums[m]])
+        return res
+
+
+so = Solution()
+print(so.threeSum([-1,0,1,2,-1,-4]))
+print(so.threeSum([0, 0, 0, 0]))
+print(so.threeSum([-2,0,0,2,2]))
+

arr.two-sum.py

@@ -0,0 +1,35 @@
+from typing import List
+
+
+class Solution:
+    """
+    1. 两数之和
+    https://leetcode-cn.com/problems/two-sum/
+    给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。
+    你可以假设每种输入只会对应一个答案。但是，数组中同一个元素不能使用两遍。
+    """
+    # hash map
+    def twoSum(self, nums: List[int], target: int) -> List[int]:
+        map = {}
+        for i, num in enumerate(nums):
+            if map.get(target - num) is not None:
+                return [map.get(target - num), i]
+            map[num] = i
+
+        return []
+
+    # 暴力破解
+    def twoSum1(self, nums: List[int], target: int) -> List[int]:
+        res = []
+        for i in range(len(nums) - 1):
+            for j in range(i + 1, len(nums)):
+                if (nums[i] + nums[j]) == target :
+                    res = [i, j]
+                    break
+
+        return res
+
+
+so = Solution()
+print(so.twoSum(nums=[2, 7, 11, 15], target=9))
+

dy.climbing-stairs.py

@@ -0,0 +1,21 @@
+
+class Solution:
+    """
+    70.爬楼梯
+    https://leetcode-cn.com/problems/climbing-stairs/
+    假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
+    每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢？
+    注意：给定 n 是一个正整数。
+    """
+    def climbStairs(self, n: int) -> int:
+        dp = {}
+        dp[1] = 1
+        dp[2] = 2
+        for i in range(3, n+1):
+            dp[i] = dp[i-1] + dp[i-2]
+
+        return dp[n]
+
+so = Solution()
+print(so.climbStairs(4))
+