C#LeetCode刷题之#496-下一个更大元素 I(Next Greater Element I)

问题

给定两个没有重复元素的数组 nums1 和 nums2 ,其中nums1 是 nums2 的子集。找到 nums1 中每个元素在 nums2 中的下一个比其大的值。

nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在,对应位置输出-1。

输入: nums1 = [4,1,2], nums2 = [1,3,4,2].

输出: [-1,3,-1]

解释:
    对于num1中的数字4,你无法在第二个数组中找到下一个更大的数字,因此输出 -1。
    对于num1中的数字1,第二个数组中数字1右边的下一个较大数字是 3。
    对于num1中的数字2,第二个数组中没有下一个更大的数字,因此输出 -1。

输入: nums1 = [2,4], nums2 = [1,2,3,4].

输出: [3,-1]

解释:
    对于num1中的数字2,第二个数组中的下一个较大数字是3。
    对于num1中的数字4,第二个数组中没有下一个更大的数字,因此输出 -1。

注意:

  • nums1和nums2中所有元素是唯一的。
  • nums1和nums2 的数组大小都不超过1000。

You are given two arrays (without duplicates) nums1 and nums2 where nums1’s elements are subset of nums2. Find all the next greater numbers for nums1's elements in the corresponding places of nums2.

The Next Greater Number of a number x in nums1 is the first greater number to its right in nums2. If it does not exist, output -1 for this number.

Input: nums1 = [4,1,2], nums2 = [1,3,4,2].

Output: [-1,3,-1]

Explanation:
    For number 4 in the first array, you cannot find the next greater number for it in the second array, so output -1.
    For number 1 in the first array, the next greater number for it in the second array is 3.
    For number 2 in the first array, there is no next greater number for it in the second array, so output -1.

Input: nums1 = [2,4], nums2 = [1,2,3,4].

Output: [3,-1]

Explanation:
    For number 2 in the first array, the next greater number for it in the second array is 3.
    For number 4 in the first array, there is no next greater number for it in the second array, so output -1.

Note:

  • All elements in nums1 and nums2 are unique.
  • The length of both nums1 and nums2 would not exceed 1000.

示意

C#LeetCode刷题之#496-下一个更大元素 I(Next Greater Element I)


示例

public class Program {

    public static void Main(string[] args) {
        var findNums = new int[] { 4, 1, 2 };
        var nums = new int[] { 1, 3, 4, 2 };

        var res = NextGreaterElement(findNums, nums);
        ShowArray(res);

        findNums = new int[] { 2, 4 };
        nums = new int[] { 1, 2, 3, 4 };

        res = NextGreaterElement2(findNums, nums);
        ShowArray(res);

        Console.ReadKey();
    }

    private static void ShowArray(int[] array) {
        foreach(var num in array) {
            Console.Write($"{num} ");
        }
        Console.WriteLine();
    }

    public static int[] NextGreaterElement(int[] findNums, int[] nums) {
        var res = new List<int>();
        var j = 0;
        var flag = false;
        var list = nums.ToList();
        for(var i = 0; i < findNums.Length; i++) {
            j = list.IndexOf(findNums[i]) + 1;
            if(j == nums.Length) {
                res.Add(-1);
                continue;
            }
            flag = false;
            while(j < nums.Length) {
                if(nums[j] > findNums[i]) {
                    res.Add(nums[j]);
                    flag = true;
                    break;
                }
                j++;
            }
            if(!flag) res.Add(-1);
        }
        return res.ToArray();
    }

    public static int[] NextGreaterElement2(int[] findNums, int[] nums) {
        //栈,请参考上方示意图
        var dic = new Dictionary<int, int>();
        var stack = new Stack<int>();
        for(var i = 0; i < nums.Length; i++) {
            dic[nums[i]] = -1;
            while(stack.Any() && nums[i] > stack.Peek()) {
                dic[stack.Pop()] = nums[i];
            }
            stack.Push(nums[i]);
        }
        var res = new List<int>();
        foreach(var num in findNums) {
            res.Add(dic[num]);
        }
        return res.ToArray();
    }

}

以上给出2种算法实现,以下是这个案例的输出结果:

-1 3 -1
3 -1

分析:

显而易见,NextGreaterElement 的时间复杂度为: C#LeetCode刷题之#496-下一个更大元素 I(Next Greater Element I) ,NextGreaterElement2 的时间复杂度为: C#LeetCode刷题之#496-下一个更大元素 I(Next Greater Element I) 。