review Palindrome

Author: gonearewe

214-最短回文串.java

@@ -1,9 +1,9 @@
 // 给定一个字符串 s，你可以通过在字符串前面添加字符将其转换为回文串。找到并返回可以用这种方式转换的最短回文串。
-
-// 示例 1:
+//
+// 示例 1:
 // 输入: "aacecaaa"
 // 输出: "aaacecaaa"
-
+//
 // 示例 2:
 // 输入: "abcd"
 // 输出: "dcbabcd"

214. 最短回文串.py

@@ -0,0 +1,38 @@
+# 给定一个字符串 s，你可以通过在字符串前面添加字符将其转换为回文串。找到并返回可以用这种方式转换的最短回文串。
+#
+# 示例 1:
+# 输入: "aacecaaa"
+# 输出: "aaacecaaa"
+#
+# 示例 2:
+# 输入: "abcd"
+# 输出: "dcbabcd"
+
+class Solution:
+    def shortestPalindrome(self, s: str) -> str:
+        def preprocess(s: str) -> str:
+            if not s:
+                return "^$"
+            T = ['^']
+            for c in s:
+                T.append('#')
+                T.append(c)
+            T.append("#$")
+            return "".join(T)
+
+        T = preprocess(s)
+        P = [0] * len(T)
+        C, R = 0, 0
+        for i in range(1, len(T) - 1):
+            i_mirror = 2 * C - i
+            P[i] = min(R - i, P[i_mirror]) if R > i else 0
+            while T[i + P[i] + 1] == T[i - P[i] - 1]:
+                P[i] += 1
+            if i + P[i] > R:
+                C, R = i, i + P[i]
+
+        end = 0
+        for i, r in enumerate(P):
+            if i - 1 == r:
+                end = max(end, (i - r) // 2 + r - 1)
+        return s[-1:end:-1] + s

42. 接雨水.py

@@ -0,0 +1,28 @@
+# 给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。
+#
+# 示例 1：
+# 输入：height = [0,1,0,2,1,0,1,3,2,1,2,1]
+# 输出：6
+# 解释：上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。
+#
+# 示例 2：
+# 输入：height = [4,2,0,3,2,5]
+# 输出：9
+#
+# 提示：
+# n == height.length
+# 0 <= n <= 3 * 104
+# 0 <= height[i] <= 105
+
+class Solution:
+    def trap(self, height: List[int]) -> int:
+        ans = 0
+        stack = []
+        for i in range(len(height) - 1, -1, -1):
+            while stack and height[stack[-1]] < height[i]:
+                h0 = height[stack.pop()]
+                if stack:
+                    h1 = min(height[i], height[stack[-1]])
+                    ans += (h1 - h0) * (stack[-1] - i - 1)
+            stack.append(i)
+        return ans