[leetcode]Next Greater Element

第一题:寻找子集合中每个元素在原集合中右边第一个比它大的数。

想到了用哈希表存这个数的位置,但是没有想到可以直接用哈希表存next great,用栈存还没找到的数,没遍历一个数就考察栈中的元素小,小的话,这个数就是栈中数的next great,栈中的数肯定是下大上小。

public int[] nextGreaterElement(int[] nums1, int[] nums2) {
         /*
        通过map建立当前元素和其next great的映射
        在建立映射时,用栈记录还没有映射(就是还没有找到next great)的数,每新遍历一个数,就考察栈顶元素能不能映射,能就
        建立映射,弹出栈顶,并继续考察新栈顶。不能建立后,压入该数。
        一开始不明白,会不会有下边小,上边大的情况,如果有这种情况的话,下边的数是建立不了映射的,但是想了想是不可能出现的,
        因为每当有大的数的时候,小的会被弹出,大的会压入,所以栈顶是最小的数。
         */
        Stack<Integer> st = new Stack<>();
        Map<Integer,Integer> map = new HashMap<>();
        int[] res = new int[nums1.length];
        for (int num :
                nums2) {
            while (!st.isEmpty() && st.peek() < num)
            {
                map.put(st.peek(),num);
                st.pop();
            }
            st.push(num);
        }
        for (int i = 0; i < res.length; i++) {
            res[i] = map.getOrDefault(nums1[i],-1);
        }
        return res;
    }

 

 

第二题:和第一题的不同点是这次是求一个循环数列中各个元素的next,而且可能有重复

循环的解决方法是遍历两轮,每次下标对n取余,这样两轮的下标就相同了

重复的解决方法是stack记录下标,而不是记录数据,每次有了next,直接存到res的相应位置

最后的res就相当于一个哈希表

public int[] nextGreaterElements2(int[] nums) {
        /*
        相对第一题的改变时数组成了循环数组,遍历到最后一个数之后可以再从第一个数开始
        自己想的方案是暴力解,两层for
        看了答案,这种循环数组遍历,下标的问题是用取余的方法,第二遍第1个数,相当于n+1个数,n+1%n = 1,正好是第一个数
        以后遇上循环数组,就遍历两倍长度,%n取余后,两次遍历的下标就一样了
         */
        int l = nums.length;
        int[] res = new int[l];
        //有的找不到,直接初始化数组为-1
        Arrays.fill(res,-1);
        Stack<Integer> st = new Stack<>();
        //循环数组,遍历两倍长度的数组,每次都取余,这样每个元素都可以把它前后的元素都遍历到,对于第二轮遍历下标不对应的解决方法是
        //取余,对n取余,下表就对应了。时间复杂度O(2n)
        //还要考虑这次有重复,所以不能记录数,要记录下标
        for (int i = 0; i < 2*l; i++) {
            int num = nums[i%l];
            while (!st.isEmpty() && nums[st.peek()] < num)
            {
                res[st.peek()] = num;
                st.pop();
            }
            //只在第一轮遍历时记录下标
            if (i<l) st.push(i);
        }
        return res;
    }

 

 

第三题:找全排列的下一个数,之前做过,但是没做出来,f**k

public int nextGreaterElement3(int n) {
        /*
        没做出来
        策略是:如果从第K为到末尾是递减的,且第k位大于第K-1位(也就是递减序列最高位是k),那么就倒序排列递减序列,并且找到
        比k-1为大的那个最小数,交换位置
        之前做过一个题,找全排列的下一个数,和这个一模一样
         */
        String str = n+"";
        int l = str.length();
        if (l == 1)
            return -1;
        //注意这里k的初始值不是0,是l-1。因为是倒着开始判断,如果没有判断到则k不会改变,递减数列应该是从l-1开始,也就是最后一个数
        int k = l-1;
        //处理成数列好操作,StringBuilder也可以
        char[] ch = str.toCharArray();
        //找到递减数列开始的地方
        for (int i = l-2; i >=0; i--) {
            if (ch[i] < ch[i+1])
            {
                k = i+1;
                break;
            }
        }
        //翻转递减数列,这里也可以转为字符串用string自带的翻转
        for (int i = k; i < k+(l-k)/2; i++) {
            char temp = ch[i];
            ch[i] = ch[l-1-(i-k)];
            ch[l-1-(i-k)] = temp;
        }
        //更换位置
        for (int i = k; i < l; i++) {
            if (ch[i] > ch[k-1])
            {
                char temp = ch[i];
                ch[i] = ch[k-1];
                ch[k-1] = temp;
                break;
            }
        }
        String res = new String(ch);
        //判断是不是超过了int最大值
        Long a = Long.parseLong(res);
        if (a >Integer.MAX_VALUE)
            return -1;
        int b = Integer.parseInt(res);
        //如果数没有改变说明这个数是全排列中最大的数,没有下一个,输出-1,要判断一下
        if (b == n)
            return -1;
        return b;
    }

 

posted @ 2018-01-04 22:19  stAr_1  阅读(239)  评论(0编辑  收藏  举报