life is too short for a diary




Next Greater Element I

Tags: leetcode python monotonic stack

Author
Written by: Tushar Sharma

The next greater element of some element x in an array is the first greater element that is to the right of x in the same array.

You are given two distinct 0-indexed integer arrays nums1 and nums2, where nums1 is a subset of nums2.

For each 0 <= i < nums1.length, find the index j such that nums1[i] == nums2[j] and determine the next greater element of nums2[j] in nums2. If there is no next greater element, then the answer for this query is -1.

Return an array ans of length nums1.length such that ans[i] is the next greater element as described above.

Brute Force

For each element in nums1, find its position in nums2, then search to the right for the first larger element.

Time Complexity: O(N2)

Monotonic Stack

A monotonic stack maintains elements in decreasing order. For each element in nums2, we track the next greater element using the stack, storing results in a hash map for O(1) lookups.

Time Complexity: O(nums1.length + nums2.length)


comments powered by Disqus