下一个更大的数 Next Greater Element

2018-09-24 21:52:38

一、Next Greater Element I

问题描述:

下一个更大的数 Next Greater Element_第1张图片

问题求解:

本题只需要将nums2中元素的下一个更大的数通过map保存下来,然后再遍历一遍nums1即可。

    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        int[] res = new int[nums1.length];
        Map map = new HashMap<>();
        Stack stack = new Stack<>();
        for (int num : nums2) {
            while (!stack.isEmpty() && stack.peek() < num) map.put(stack.pop(), num);
            stack.push(num);
        }
        for (int i = 0; i < nums1.length; i++) {
            res[i] = map.getOrDefault(nums1[i], -1);
        }
        return res;
    }

 

二、Next Greater Element II

问题描述:

下一个更大的数 Next Greater Element_第2张图片

问题求解:

本题和上一题应该来说是差不多的,都可以使用Stack来对数据进行维护,有两个需要注意的地方,一个是本题中的循环性,解决循环性的方法就是在数组后再加上一个数组即可。

二一个是本题中并没有说是无重复元素的,因此只能在栈中保存index值。

栈中保存的只有第一次遍历的下标值,第二次就没有必要再进行压栈操作了,另外,栈中保存的是递减的数的index。

    public int[] nextGreaterElements(int[] nums) {
        int[] res = new int[nums.length];
        Arrays.fill(res, -1);
        Stack stack = new Stack<>();
        int n = nums.length;
        for (int i = 0; i < 2 * n; i++) {
            int num = nums[i % n];
            while (!stack.isEmpty() && nums[stack.peek()] < num) res[stack.pop()] = num;
            if (i < n) stack.push(i);
        }
        return res;
    }

 

三、Next Greater Element III

问题描述:

下一个更大的数 Next Greater Element_第3张图片

问题求解:

就是一条按字典序的下一个Permutation。

    public int nextGreaterElement(int n) {
        char[] c = String.valueOf(n).toCharArray();
        int idx = c.length - 2;
        while (idx >= 0 && c[idx] >= c[idx + 1]) idx--;
        if (idx == -1) return -1;
        int j = c.length - 1;
        while (j >= 0 && c[j] <= c[idx]) j--;
        swap(c, idx, j);
        reverseSort(c, idx + 1, c.length - 1);
        long res = Long.valueOf(String.valueOf(c));
        return res >= Integer.MAX_VALUE ? -1 : (int)res;
    }

    private void swap(char[] c, int i, int j) {
        char tmp = c[i];
        c[i] = c[j];
        c[j] = tmp;
    }
    
    private void reverseSort(char[] c, int start, int end) {
        if (start >= end) return;
        while (start < end) {
            swap(c, start, end);
            start++;
            end--;
        }
    }

 

转载于:https://www.cnblogs.com/hyserendipity/p/9696911.html

你可能感兴趣的:(下一个更大的数 Next Greater Element)