1.0.16 更新了 Optional[TreeNode]

Author: KuiyuanFu

CHANGELOG.md

@@ -23,3 +23,5 @@ All notable changes to the "leetcode" extension will be documented in this file.
 更新了 imports.py random
 # 1.0.15
 更新了 imports.py random fix
+# 1.0.16
+更新了 Optional[TreeNode]

package.json

@@ -2,7 +2,7 @@
     "name": "python-leetcode",
     "displayName": "PythonLeetcode",
     "description": "Solve LeetCode problems in VS Code",
-    "version": "1.0.15",
+    "version": "1.0.16",
     "author": "KuiyuanFu",
     "publisher": "KuiyuanFu",
     "license": "MIT",

src/leetCodeExecutor.ts

@@ -346,10 +346,10 @@ class LeetCodeExecutor implements Disposable {
                     p = p.slice(0, p.length - 1)
                 }
 
-                if (paraItemPair[1] == 'ListNode') {
+                if (paraItemPair[1] == 'ListNode'|| paraItemPair[1] == 'Optional[ListNode]') {
                     return "listToListNode(" + p + ")"
                 }
-                else if (paraItemPair[1] == 'TreeNode') {
+                else if (paraItemPair[1] == 'TreeNode' || paraItemPair[1] == 'Optional[TreeNode]') {
                     p = p.replace(/null/g, 'None')
                     return "listToTreeNode(" + p + ")"
                 }