Merge pull request #2711 from JasenWn/master

更新0503.下一个更大元素II的python版本

Author: youngyangyang04

problems/0503.下一个更大元素II.md

@@ -168,6 +168,7 @@ class Solution {
 ```
 
 ### Python: 
+> 版本一：
 
 ```python
 class Solution:
@@ -181,6 +182,34 @@ class Solution:
             stack.append(i%len(nums))
         return dp
 ```
+
+> 版本二：针对版本一的优化
+
+```python3
+class Solution:
+    def nextGreaterElements(self, nums: List[int]) -> List[int]:
+        res = [-1] * len(nums)
+        stack = []
+        #第一次遍历nums
+        for i, num in enumerate(nums):   
+            while stack and num > nums[stack[-1]]:
+                res[stack[-1]] = num
+                stack.pop()
+            stack.append(i)
+        #此时stack仍有剩余，有部分数‘无下一个更大元素’待修正
+        #第二次遍历nums
+        for num in nums:
+            #一旦stack为空，就表明所有数都有下一个更大元素，可以返回结果    
+            if not stack:   
+                return res
+            while stack and num > nums[stack[-1]]:
+                res[stack[-1]] = num
+                stack.pop()
+            #不要将已经有下一个更大元素的数加入栈，这样会重复赋值，只需对第一次遍历剩余的数再尝试寻找下一个更大元素即可
+        #最后仍有部分最大数无法有下一个更大元素，返回结果
+        return res  
+```
+
 ### Go:
 
 ```go
@@ -203,7 +232,6 @@ func nextGreaterElements(nums []int) []int {
     return result
 }
 ```
-
 ### JavaScript:
 
 ```JS