我是靠谱客的博主 香蕉月光,最近开发中收集的这篇文章主要介绍【Lintcode】1877. Longest Super Sequence题目地址:,觉得挺不错的,现在分享给大家,希望可以做个参考。

概述

题目地址:

https://www.lintcode.com/problem/longest-super-sequence/description

给定一个长 n n n数组 A A A,要求找到其最长子数组,删掉其一个元素之后别的元素出现次数都一样。返回这个子数组。题目保证 A A A里只含 0 ∼ 3000 0sim 3000 03000

思路是枚举“别的元素出现次数都一样”这个出现次数 x x x。先统计一下每个数字的出现次数,接着开个最小堆,将出现了的数入堆,出现次数小的优先级高,并且存一下出现次数最多的那个数是谁。我们的目的是,每次从堆中弹出一个数 a a a的时候,如果其出现次数是 c c c,那么我们枚举 x = c − 1 x=c-1 x=c1 x = c x=c x=c的情况。假设poll出 a a a之后堆的size是 s s s x = c − 1 x=c-1 x=c1的情况是指, a a a以及堆里的数都重复 c − 1 c-1 c1次,另外再加上 a a a自己,此时整个LSS的长度是 ( c − 1 ) ( s + 1 ) + 1 (c-1)(s+1)+1 (c1)(s+1)+1 x = c x=c x=c的情况是, a a a以及堆里的数都重复 c c c次,再加上一个堆里的某个数多一次,此时整个LSS的长度是 c ( s + 1 ) + 1 c(s+1)+1 c(s+1)+1,这里有个必要条件是 A A A里存在某个数出现次数超过 c c c,否则无法存在这个数,例如 A A A里每个数都出现了 2 2 2次,则 c c c是无法枚举为 2 2 2的,因为没有数次数比 2 2 2多。但是此时 c c c仍然可以枚举为 1 1 1。这里还需要注意 A A A里所有数都只出现了 1 1 1次的情况,这个情况不含在上面的讨论中,此时直接返回 A A A自己。代码如下:

import java.util.Arrays;
import java.util.PriorityQueue;

public class Solution {
    /**
     * @param nums: the input array
     * @return: return the longest super sequence
     */
    public int[] longestSuperSequence(int[] nums) {
        // write your code here
        int[] count = new int[3001];
        int maxCount = 0, maxCountNum = 0;
        // 统计一下每个数出现次数,以及出现次数最多的数是几,及其出现次数
        for (int num : nums) {
            count[num]++;
            if (count[num] > maxCount) {
                maxCount = count[num];
                maxCountNum = num;
            }
        }
        
        // 特判一下出现次数最多的元素只出现一次的情况
        if (maxCount == 1) {
            return nums;
        }
        
        // 建堆,把出现过的数入堆,出现次数少的数优先
        PriorityQueue<Integer> minHeap = new PriorityQueue<>((x, y) -> Integer.compare(count[x], count[y]));
        for (int i = 0; i < count.length; i++) {
            if (count[i] > 0) {
                minHeap.offer(i);
            }
        }
        
        // maxLen是可以达到的最长LSS的长度,mostCount是这个LSS去掉一个数后别的数出现次数一样了,这个出现次数
        int maxLen = 0, mostCount = 0;
        while (!minHeap.isEmpty()) {
            int cur = minHeap.poll(), curCount = count[cur];
            if ((count[cur] - 1) * (minHeap.size() + 1) + 1 > maxLen) {
                maxLen = (count[cur] - 1) * (minHeap.size() + 1) + 1;
                mostCount = count[cur] - 1;
            }
            
            if (count[cur] * (minHeap.size() + 1) + 1 > maxLen && maxCount > count[cur]) {
                maxLen = count[cur] * (minHeap.size() + 1) + 1;
                mostCount = count[cur];
            }
            
            // 这里可以做个优化,一旦枚举完这个curCount,就把所有出现次数也是curCount的数全poll出来
            while (!minHeap.isEmpty() && count[minHeap.peek()] == curCount) {
                minHeap.poll();
            }
        }
        
        int[] res = new int[maxLen], resCount = new int[3001];
        int idx = 0;
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] == maxCountNum && resCount[maxCountNum] < mostCount + 1) {
                res[idx++] = nums[i];
                resCount[nums[i]]++;
            } else if (nums[i] != maxCountNum && count[nums[i]] >= mostCount && resCount[nums[i]] < mostCount) {
                res[idx++] = nums[i];
                resCount[nums[i]]++;
            }
        }
        
        return res;
    }
}

时间复杂度 O ( n log ⁡ n ) O(nlog n) O(nlogn),空间 O ( n ) O(n) O(n)

最后

以上就是香蕉月光为你收集整理的【Lintcode】1877. Longest Super Sequence题目地址:的全部内容,希望文章能够帮你解决【Lintcode】1877. Longest Super Sequence题目地址:所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(40)

评论列表共有 0 条评论

立即
投稿
返回
顶部