committed from zkp

pythonkd · pythonkd · commit 8f1679abef4e · 2019-03-12T15:53:41.000+08:00
diff --git a/LeetCode/nextGreaterElement-1.py b/LeetCode/nextGreaterElement-1.py
@@ -0,0 +1,15 @@
+from collections import deque
+class Solution(object):
+    def nextGreaterElement(self, nums1, nums2):
+        """
+        :type nums1: List[int]
+        :type nums2: List[int]
+        :rtype: List[int]
+        """
+        s = deque()
+        d = {}
+        for i in nums2:
+            while s and s[-1] < i:
+                d[s.pop()] = i
+            s.append(i)
+        return [d[i] if i in d else -1 for i in nums1]
